【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;
}