1. 程式人生 > >PACM Team(多重揹包+路徑記錄)

PACM Team(多重揹包+路徑記錄)

正文:

所謂多重揹包,就是在其他屬性上有一些限制,除了空間可能還有其他的限制。這個其實就是幾個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; } } }