PACM Team(多重揹包+路徑記錄)
阿新 • • 發佈:2018-12-13
正文:
所謂多重揹包,就是在其他屬性上有一些限制,除了空間可能還有其他的限制。這個其實就是幾個for迴圈掃一下就好了,幾個限制就開幾維陣列。 比如此題有四個限制 ,那麼就開四維陣列就好了 其中ma的意思是maxa a限制的最大值,mb、mc、md,同理。
for(int i =0;i<n;i++){
for(int a1 = ma;a1>=a[i];a1--){
for(int b1 = mb;b1>=b[i];b1--){
for(int c1 = mc;c1>=c[i];c1--){
for (int d1 = md;d1>=d[i];d1--){
dp[a1][b1][c1][d1] = max(dp[a1][b1][c1][d1],dp[a1-a[i]][b1-b[i]][c1-c[i]][d1-d[i]]+val[i]);
}
}
}
}
}
題目大意: 有許多隊伍,每個隊伍4種專家,要麼全部都要,要麼一個都不要,每種專家有人數限制,每個隊伍有一個值,問在給定限制內如何讓總值最大。 簡單的dp問題。設一個4維陣列表示4個專家,每個隊伍4層迴圈走01揹包。其中的難點主要在於要求輸出取用了那些隊伍。由於隊伍總數才36,2的36次方可以用longlong存下,所以用二進位制儲存的方法即可。
int tmp = dp[a1-a[i]][b1-b[i]][c1-c[i]][d1-d[i]]+val[i];
if(dp[a1][b1][c1][d1]<tmp){
dp[a1][b1][c1][d1] = tmp;
LL wei = 1;
use[a1][b1][c1][d1] = (use[a1-a[i]][b1-b[i]][c1-c[i]][d1-d[i]]|(LL)(wei<<i));
}
放進vector輸出即可 AC程式碼:
#include<bits/stdc++.h>
#define LL long long
using namespace std;
int dp[40][40][40][40];
LL use[40][40][40][40];
int a[40];
int b[40];
int c[40];
int d[40];
int val[40];
int ma,mb,mc,md;
int main(){
int n;
while(~scanf("%d",&n)){
memset(dp,0,sizeof dp);
memset(use,0,sizeof use);
for(int i = 0;i<n;i++){
scanf("%d%d%d%d%d",&a[i],&b[i],&c[i],&d[i],&val[i]);
}
scanf("%d%d%d%d",&ma,&mb,&mc,&md);
for(int i =0;i<n;i++){
for(int a1 = ma;a1>=a[i];a1--){
for(int b1 = mb;b1>=b[i];b1--){
for(int c1 = mc;c1>=c[i];c1--){
for(int d1 = md;d1>=d[i];d1--){
int tmp = dp[a1-a[i]][b1-b[i]][c1-c[i]][d1-d[i]]+val[i];
if(dp[a1][b1][c1][d1]<tmp){
dp[a1][b1][c1][d1] = tmp;
LL wei = 1;
use[a1][b1][c1][d1] = (use[a1-a[i]][b1-b[i]][c1-c[i]][d1-d[i]]|(LL)(wei<<i));
}
}
}
}
}
}
queue<int>ans;
while(!ans.empty()){
ans.pop();
}
LL tmp = use[ma][mb][mc][md];
//cout<<tmp<<"t"<<endl;
int cnt = 0;
for(int i =0;i<36;i++){
if(tmp%2==1){
ans.push(i);
cnt++;
}
tmp/=2;
}
printf("%d\n",cnt);
if(cnt!=0){
int x =0;
while(!ans.empty()){
if(x!=0)printf(" ");
printf("%d",ans.front());
ans.pop();
x++;
}
cout<<endl;
}
}
}