救命

P1064 [NOIP2006 提高组] 金明的预算方案

cts0219 @ 2023-11-19 10:12:49

大红大紫的,30pts,求调

#include <bits/stdc++.h>
using namespace std;
const int N=3.2e4+5;
int n,m,dp[N];
int zj[65][3],fj[65][3][3];
int main()
{
    cin>>n>>m;
    for (int i=1;i<=n;i++)
    {
        int v,p,q;
        cin>>v>>p>>q;
        if (q==0)
        {
            zj[i][1]=v;
            zj[i][2]=v*p;
        }
        else
        {
            if (fj[q][1][1]==0)
            {
                fj[q][1][1]=v;
                fj[q][1][2]=v*p;
            }
            else
            {
                fj[q][2][1]=v;
                fj[q][2][2]=v*p;
            }
        }
    }
    for (int i=1;i<=m;i++)
    {
        if (zj[i][1]==0) continue;
        for (int j=n;j>=zj[i][1];j--)
        {
            dp[j]=max(dp[j],dp[j-zj[i][1]]+zj[i][2]);
            if (fj[i][1][1]!=0&&j>=zj[i][1]+fj[i][1][1])
                dp[j]=max(dp[j],dp[j-zj[i][1]-fj[i][1][1]]+zj[i][2]+fj[i][1][2]);
            if (fj[i][2][1]!=0&&j>=zj[i][1]+fj[i][2][1])
                dp[j]=max(dp[j],dp[j-zj[i][1]-fj[i][2][1]]+zj[i][2]+fj[i][2][2]);
            if (fj[i][1][1]!=0&&fj[i][2][1]!=0&&j>=zj[i][1]+fj[i][1][1]+fj[i][2][1])
                dp[j]=max(dp[j],dp[j-zj[i][1]-fj[i][1][1]-fj[i][2][1]]+zj[i][2]+fj[i][1][2]+fj[i][2][2]);
        }
    }
    cout<<dp[n];
}

|