洛谷 P2623 物品选取

https://www.luogu.org/problemnew/show/P2623

https://www.luogu.org/blog/test-1/solution-p2623

重点就是甲类物品最多取一个，一定能取到最优解。。。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
#define fi first
#define se second
#define mp make_pair
#define pb push_back
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> pll;

ll ans[2005],an2;
//前i个物品，用j的容量的最大价值
vector<pll> dd;
ll n,m;
int main()
{
    ll idx,i,j,k,a,b,c;
    scanf("%lld%lld",&n,&m);
    for(i=1;i<=n;i++)
    {
        scanf("%lld",&idx);
        if(idx==1)
        {
            scanf("%lld%lld",&a,&b);
            dd.pb(mp(a,b));
        }
        else if(idx==2)
        {
            scanf("%lld%lld%lld",&a,&b,&c);
            if(b==0)
            {
                for(j=0;j<=m;j++)    ans[j]+=a*c;
            }
            else
            {
                for(j=m;j>=1;j--)
                {
                    for(k=min(c,j/b);k>=1;k--)
                    {
                        ans[j]=max(ans[j],ans[j-k*b]+k*a);
                    }
                }
            }
        }
        else if(idx==3)
        {
            scanf("%lld%lld",&a,&b);
            if(b==0)    exit(-1);
            else
            {
                for(j=b;j<=m;j++)
                {
                    ans[j]=max(ans[j],ans[j-b]+a);
                }
            }
        }
        //for(int i=0;i<=m;i++)    printf("a%lld %lld
",i,ans[i]);
    }
    for(i=1;i<=m;i++)    ans[i]=max(ans[i],ans[i-1]);
    an2=ans[m];
    for(i=0;i<dd.size();i++)
    {
        for(j=0;j<=m;j++)
        {
            an2=max(an2,ans[m-j]+dd[i].fi*j*j-dd[i].se*j);
        }
    }
    printf("%lld",an2);
    return 0;
}