求助!

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

沧海梦 @ 2019-03-04 23:02:02

基础dp写了两个多小时,dp方程应该没问题,但是始终有三个点过不去,求大佬赐教!

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int dp[65][33000],w[80][3],c[80][3];
int n,m,v,p,q,k=1;
int ans;
int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>v>>p>>q;
        if(!q) w[k][0]=p,c[k++][0]=v;
        else if(!w[q][1]) w[q][1]=p,c[q][1]=v;
        else w[q][2]=p,c[q][2]=v;
    }
    for(int i=1;i<=k;i++)
    for(int j=0;j<=n;j++){

        if(j-c[i][0]>=0){
            dp[i][j]=max(dp[i-1][j-c[i][0]]+w[i][0]*c[i][0],dp[i-1][j]);
            if(c[i][1]&&j-c[i][0]-c[i][1]>=0)
            dp[i][j]=max(dp[i-1][j-c[i][0]-c[i][1]]+w[i][0]*c[i][0]+w[i][1]*c[i][1],dp[i-1][j]);
            if(c[i][2]&&j-c[i][0]-c[i][2]>=0)
            dp[i][j]=max(dp[i-1][j-c[i][0]-c[i][2]]+w[i][0]*c[i][0]+w[i][2]*c[i][2],dp[i-1][j]);
            if(c[i][1]&&c[i][2]&&j-c[i][0]-c[i][1]-c[i][2]>=0)
            dp[i][j]=max(dp[i-1][j-c[i][0]-c[i][1]-c[i][2]]+w[i][0]*c[i][0]+w[i][1]*c[i][1]+w[i][2]*c[i][2],dp[i-1][j]);

        }
        dp[i][j]=max(dp[i][j],dp[i-1][j]);
        ans=max(ans,dp[i][j]);

    }
    cout<<ans;

    return 0;
}

|