Newcoder 141 A.PACM Team（01背包）

Description

有四种类型花费一种收益的$n$件物品，问在四种类型花费不超过各自上限的最大收益

Input

第一行一整数$n$表示物品数量，之后$n$行每行五个整数$p_i,a_i,c_i,m_i,g_i$分别表示该物品的四种花费和收益，最后输入四个整数$P,A,C,M$表示四种花费的上限

$(1\le n\le 36,0\le p_i,a_i,c_i,m_i,g_i,P,A,C,M\le 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$背包，内存稍大，用$short$即可

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;
}