题 意:小明想参加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
*/