Description
有四种类型花费一种收益的 n n 件物品,问在四种类型花费不超过各自上限的最大收益
Input
第一行一整数表示物品数量,之后 n n 行每行五个整数分别表示该物品的四种花费和收益,最后输入四个整数 P,A,C,M P , A , C , M 表示四种花费的上限
(1≤n≤36,0≤pi,ai,ci,mi,gi,P,A,C,M≤36) ( 1 ≤ n ≤ 36 , 0 ≤ p i , a i , c i , m i , g i , P , A , C , M ≤ 36 )
Output
输出能够获得最大收益所选的物品数量和这些物品的编号
Sample Input
2
1 0 2 1 10
1 0 2 1 21
1 0 2 1
Sample Output
1
1
Solution
01 01 背包,内存稍大,用 short s h o r t 即可
Code
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
short n,P,A,C,M,p[37],a[37],c[37],m[37],g[37],dp[37][37][37][37][37];
vector<int>v;
void dfs(int i,int x,int y,int z,int w)
{
if(i==0)return ;
if(dp[i-1][x][y][z][w]==dp[i][x][y][z][w])dfs(i-1,x,y,z,w);
else
{
v.push_back(i);
dfs(i-1,x-p[i],y-a[i],z-c[i],w-m[i]);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d%d%d%d%d",&p[i],&a[i],&c[i],&m[i],&g[i]);
scanf("%d%d%d%d",&P,&A,&C,&M);
for(int i=0;i<n;i++)
for(int x=0;x<=P;x++)
for(int y=0;y<=A;y++)
for(int z=0;z<=C;z++)
for(int w=0;w<=M;w++)
{
dp[i+1][x][y][z][w]=max(dp[i+1][x][y][z][w],dp[i][x][y][z][w]);
if(x+p[i+1]<=P&&y+a[i+1]<=A&&z+c[i+1]<=C&&w+m[i+1]<=M)
dp[i+1][x+p[i+1]][y+a[i+1]][z+c[i+1]][w+m[i+1]]=max(dp[i+1][x+p[i+1]][y+a[i+1]][z+c[i+1]][w+m[i+1]],(short)(dp[i][x][y][z][w]+g[i+1]));
}
short ans=0;
for(int x=0;x<=P;x++)
for(int y=0;y<=A;y++)
for(int z=0;z<=C;z++)
for(int w=0;w<=M;w++)
ans=max(ans,dp[n][x][y][z][w]);
int flag=1;
for(int x=0;x<=P&&flag;x++)
for(int y=0;y<=A&&flag;y++)
for(int z=0;z<=C&&flag;z++)
for(int w=0;w<=M&&flag;w++)
if(ans==dp[n][x][y][z][w])
{
flag=0;
dfs(n,x,y,z,w);
}
printf("%d\n",v.size());
for(int i=0;i<v.size();i++)printf("%d%c",v[i]-1,i==v.size()-1?'\n':' ');
return 0;
}