沧海梦 @ 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;
}