[BZOJ2794][Poi2012]Cloakroom

2794: [Poi2012]Cloakroom

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 167  Solved: 119
[Submit][Status][Discuss]

Description

有n件物品，每件物品有三个属性a[i], b[i], c[i] (a[i]<b[i])。
再给出q个询问，每个询问由非负整数m, k, s组成，问是否能够选出某些物品使得：
1. 对于每个选的物品i，满足a[i]<=m且b[i]>m+s。
2. 所有选出物品的c[i]的和正好是k。

Input

第一行一个正整数n (n<=1,000)，接下来n行每行三个正整数，分别表示c[i], a[i], b[i] (c[i]<=1,000, 1<=a[i]<b[i]<=10^9)。
下面一行一个正整数q (q<=1,000,000)，接下来q行每行三个非负整数m, k, s (1<=m<=10^9, 1<=k<=100,000, 0<=s<=10^9)。

Output

输出q行，每行为TAK (yes)或NIE (no)，第i行对应第i此询问的答案。

Sample Input

5

6 2 7

5 4 9

1 2 4

2 5 8

1 3 9

5

2 7 1

2 7 2

3 2 0

5 7 2

4 1 5

Sample Output

TAK

NIE

TAK

TAK

NIE

HINT

 

Source

鸣谢Oimaster

[Submit][Status][Discuss]

先看数据范围，发现q非常的大，所以想到离线处理。

如果离线处理，可以把a的这个条件直接去掉，只需用b和c来DP。

然后就非常简单了。时间复杂度$O(qlogq+nk)$

 #include<cstdio>
 #include<algorithm>
 #define N 1005
 #define M 100010
 #define Q 1000100
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 int n,q,f[M]={1e9},ans[Q],maxn;
 struct node{int a,b,c,id;}a[N],b[Q];
 bool operator<(node x,node y){return x.a<y.a;}
 int main()
 {
     n=read();
     for(int i=;i<=n;i++)
     a[i].c=read(),a[i].a=read(),a[i].b=read();
     sort(a+,a+n+);
     q=read();
     for(int i=;i<=q;i++)
     b[i].a=read(),b[i].c=read(),b[i].b=read(),
     b[i].id=i,maxn=max(maxn,b[i].c);
     sort(b+,b+q+);
     for(int i=,j=;i<=q;i++)
     {
         while(j<=n&&a[j].a<=b[i].a)
         {
             for(int k=maxn;k>=a[j].c;k--)
             f[k]=max(f[k],min(f[k-a[j].c],a[j].b));
             j++;
         }
         ans[b[i].id]=(f[b[i].c]>b[i].a+b[i].b);
     }
     for(int i=;i<=q;i++)
     puts(ans[i]?"TAK":"NIE");
 }