【JZOJ junior 2059】 稀有逛超市

Description

稀有打算去超市买点东西，他身上共有 N 元钱，其中 K 元钱都是准备用来买游戏打的， 因此十分蛋疼的就是剩下的钱安排买什么了。超市的每件东西有很多类，第一类只有价格和 给稀有带来的愉悦值，但是每种只有一件；第二类每种有无数件，同样有价格和愉悦值；其 他类除了每一种有价格和愉悦值之外，每一类一共只能取一件。（我会告诉你稀有去了 HZ 之后身上没钱了？）
对于 50%的数据，只含前两类物品。
对于 100%的数据，K<=N<=20000，M<=1000，|Wi|<=10000，|Pi|<=10000。
最多一共有 1000 类物品，类型最大为第 1000 类。

Analysis

出题人的语文非常好。。。。我的语文也非常好。。。
三种背包强行加在一起啊，每种分别dp，至于限制可以在循环时用downto解决。
难点就是限制的解决咯，其实好好学01背包的一维数组版本的都很容易想到。

Code

#include<cstdio>#include<algorithm>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fd(i,b,a) for(int i=b;i>=a;i--)using namespace std;const int N=1010,M=20010;int n,m,na,nb,nc,f[M];struct node{    int p,w,c;}alan,a[N],b[N],c[N];bool cmp(node a,node b){    return a.c<b.c;}int main(){    freopen("shop.in","r",stdin);    freopen("shop.out","w",stdout);    int _;    scanf("%d %d %d",&m,&_,&n);    m-=_;    fo(i,1,n)    {        scanf("%d %d %d",&alan.p,&alan.w,&alan.c);        if(alan.c==1) a[++na]=alan;        else        if(alan.c==2) b[++nb]=alan;        else c[++nc]=alan;    }    fo(i,1,na)        fd(j,m,a[i].p) f[j]=max(f[j],f[j-a[i].p]+a[i].w);    fo(i,1,nb)        fo(j,b[i].p,m) f[j]=max(f[j],f[j-b[i].p]+b[i].w);    int l=1;    sort(c+1,c+nc+1,cmp);    fo(i,2,nc+1)        if(c[i].c!=c[i-1].c)        {            fd(j,m,0)                fo(k,l,i-1)                {                    if(j-c[k].p<0) continue;                    f[j]=max(f[j],f[j-c[k].p]+c[k].w);                }            l=i;        }    int ans=0;    fo(i,0,m) ans=max(ans,f[i]);    printf("%d",ans);    return 0;}