牛客多校训练赛第三场 A PACM Team

A PACM Team

题 意:小明想参加icpc，有n个队伍可以选择。每个队伍包括搞physics 的p人，搞Algorithm的a人，搞Coding的c人，搞math的m人，能给小明带来的知识点为g。
小明还希望队伍搞同一种知识点的人不要太多。于是希望选出之后，队伍里面搞Coding的人不要超过C人，math的不要超过M人,搞physics的人不要超过P人，搞Algorithm的人不要超过A人。求在不超过限制的情况下，小明选的队伍知识点总和最多。
数据范围:
1<=N<=36
0<=pi,ai,ci,mi,gi<=36
0<=P,A,C，M,G<=36
输入样例:

3
1 0 2 1 10
1 0 2 1 21
1 0 2 1 100
2 0 3 2

输出样例:

1
2

思 路:就算一个01背包，加路径还原，五纬的。卡注意内存

#include<bits/stdc++.h>
using namespace std;
const int maxn = 37;
typedef long long LL;
short P[maxn],A[maxn],C[maxn],M[maxn],G[maxn];
short dp[maxn][maxn][maxn][maxn][maxn];
bool last[maxn][maxn][maxn][maxn][maxn];
short n,lP,lA,lC,lM,lG;
bool judge(int i,int j,int k,int p,int q){
    if(j >= P[i] && k>=A[i] && p>=C[i] && q>= M[i]) return true;
    else return false;
}
void Fu(int i,int j,int k,int p,int q){
    short temp = dp[i-1][j-P[i]][k-A[i]][p-C[i]][q-M[i]]+G[i];
    if(temp > dp[i][j][k][p][q]){
        dp[i][j][k][p][q] = temp;
        last[i][j][k][p][q] = true;
    }
}
int main(){
    scanf("%ld",&n);
    for(int i=1;i<=n;i++){
        scanf("%ld %ld %ld %ld %ld",&P[i],&A[i],&C[i],&M[i],&G[i]);
    }
    scanf("%ld %ld %ld %ld",&lP,&lA,&lC,&lM);
    memset(dp,0,sizeof(dp));
    memset(last,0,sizeof(last));
    for(int i=1;i<=n;i++){
        for(int j=0;j<=lP;j++){
            for(int k=0;k<=lA;k++){
                for(int p=0;p<=lC;p++){
                    for(int q=0;q<=lM;q++){
                        dp[i][j][k][p][q] = dp[i-1][j][k][p][q];
                        if(judge(i,j,k,p,q)){
                            Fu(i,j,k,p,q);
                        }
                    }
                }
            }
        }
    }

    int ans = 0;
    int step= n;
    int p1=0,a1=0,c1=0,m1=0;
    for(int j=0;j<=lP;j++){
        for(int k=0;k<=lA;k++){
            for(int p=0;p<=lC;p++){
                for(int q=0;q<=lM;q++){
                    if(dp[n][j][k][p][q] > ans){
                        ans = dp[n][j][k][p][q];
                        p1=j;a1=k;c1=p;m1=q;
                    }
                }
            }
        }
    }

    vector<int> vec;
    vec.clear();
    while(step>=1 && p1>=0 && a1>=0 && c1>=0 && m1>=0){
        if(last[step][p1][a1][c1][m1]){
            vec.push_back(step);
            p1 -= P[step];a1-=A[step];
            c1 -= C[step];m1-=M[step];
        }
        step--;
    }
    reverse(vec.begin(),vec.end());
    printf("%d\n",vec.size());
    for(int i=0;i<vec.size();i++){
        printf("%d%c",vec[i]-1,i==vec.size()-1?'\n':' ');
    }
    return 0;
}
/*
3
1 0 2 1 10
1 0 2 1 21
1 0 2 1 100
2 0 3 2

1
2 1 1 0 31
2 1 2 1
*/