多重背包問題


多重背包問題

給定\(n\)種物品,第\(i\)種共有\(c_i\)個,價值為\(v_i\),重量為\(w_i\)。現在有一個背包,最大載重量為\(m\)。求若選一些物品放到背包里,最多能放的總價值是多少。

解法\(1\)

考慮將多重背包轉化為01背包。最簡單的想法是將\(1\)種物品直接拆分成\(c_i\)個相同的物品,然后01背包。這樣時間復雜度是\(\mathrm O\!\left(\sum\limits_{i=1}^nc_i\cdot m\right)\),太大了。考慮有沒有更優的拆分方式。

我們先看這樣一個問題:給定一個數\(x\),問最少能選多少個數,使得\(\left[0,2^x\right)\)內每個數都能被表示成一些選出的數之和。很顯然可以選\(2^0,2^1,\cdots,2^{x-1}\)\(x\)個數,利用的是任何數都可以被二進制拆分這個原理。那么如果給定一個數\(x\),問的是最少能選多少個數,使得\([0,x]\)內每個數都能被表示成一些選出的數之和呢?考慮找出\(x\)以內(包括\(x\))最大的一個能被表示成\(2\)的整次冪\(-1\)的數\(2^y-1\),那么可以選\(y\)個數使得\(\left[0,2^y\right)\)內每個數都能被表示成一些選出的數之和(顯然\(y=\lfloor\log_2(x+1)\rfloor\))。那么對於\(\left[2^y,x\right]\)內的數呢?只需要再添一個數\(x-2^y+1\)即可。因為\(\forall i\in\left[2^y,x\right]\),顯然\(i-\left(x-2^y+1\right)\in\left[2^{y+1}-x-1,2^y-1\right]\subseteq\left[0,2^y\right)\),那么我們可以先湊出\(i-\left(x-2^y+1\right)\),再加一個\(x-2^y+1\)上去。

現在回到多重背包問題。第\(i\)種物品顯然可以被這樣拆分:\((2^0v_i,2^0w_i),(2^1v_i,2^1w_i),\cdots,\left(2^{\lfloor\log_2(c_i+1)\rfloor-1}v_i,2^{\lfloor\log_2(c_i+1)\rfloor-1}w_i\right),\left(\left(c_i-2^{\lfloor\log_2(c_i+1)\rfloor}+1\right)v_i,\left(c_i-2^{\lfloor\log_2(c_i+1)\rfloor}+1\right)w_i\right)\)(其中\((x,y)\)表示一個價值為\(x\),重量為\(y\)的物品)。這樣當且僅當\(j\in[0,c_i]\)時,\(j\)個第\(i\)種物品能被等價地表示出來,並且拆分成的物品的數量是\(\log\)級別的。於是這樣拆分完再01背包,復雜度就有保證了:\(\mathrm O\!\left(\sum\limits_{i=1}^n\log c_i\cdot m\right)\)。空間復雜度為拆分出來的物品個數+01背包所需空間:\(\mathrm O\!\left(\sum\limits_{i=1}^n\log c_i+m\right)\)

下面貼代碼:

int nwn/*新物品個數*/,nwv[N*LOG_C_I+1]/*新物品價值*/,nww[N*LOG_C_I+1]/*新物品重量*/;
int dp[M+1];

nwn=0; 
for(int i=1;i<=n;i++){//拆分每種物品 
	int _log=log2(c[i]+1); 
	for(int j=0;j<_log;j++)nwv[++nwn]=v[i]<<j,nww[nwn]=w[i]<<j;//湊[0,2^_log) 
	if(c[i]>(1<<_log)-1)nwv[++nwn]=v[i]*(c[i]-(1<<_log)+1),nww[nwn]=w[i]*(c[i]-(1<<_log)+1);//湊[2^_log,c[i]] 
}
memset(dp,0,sizeof(dp));
for(int i=1;i<=nwn;i++)for(int j=m;j>=nww[i];j--)//01背包 
	dp[j]=max(dp[j],dp[j-nww[i]]+nwv[i]);
//目標為dp[m]

解法\(2\)

直接DP。設\(dp_{i,j}\)表示當前考慮到第\(i\)個物品,背包容量還剩\(j\)時能放的最大價值。考慮枚舉第\(i\)個物品選了多少個,很容易得到轉移方程\(dp_{i,j}=\max\limits_{k=0}^{\min\left(c_i,\left\lfloor\frac j{w_i}\right\rfloor\right)}\left\{dp_{i-1,j-kw_i}+kv_i\right\}\)。這個方程不管是列法上還是長相上都一臉單調隊列的樣子。於是我們將它變變形,分離狀態變量\(j\)和決策變量\(k\)(其實就是改為枚舉剩余容量),得\(dp_{i,j}=\max\limits_{k\in[\max(0,j-c_iw_i),j],k\equiv j\!\pmod{w_i}}\left\{dp_{i-1,k}-\dfrac k{w_i}v_i+\dfrac j{w_i}v_i\right\}\)。考慮到這里面運算時會有小數,於是我們先加減后除,得\(dp_{i,j}=\max\limits_{k\in[\max(0,j-c_iw_i),j],k\equiv j\!\pmod{w_i}}\left\{\dfrac{w_idp_{i-1,k}-kv_i+jv_i}{w_i}\right\}\)

這樣就很顯然怎么單調隊列了吧。注意到\(\max\)的條件里有一個同余,於是我們可以把狀態變量\(j\)分為\(w_i\)組,每組中的成員分別\(\bmod w_i=0,1,\cdots,w_i-1\),每組單獨跑一遍單調隊列,維護當前狀態的所有決策中使\(w_idp_{i-1,k}-kv_i\)最大的決策\(k\)。而每到達一個\(j\),將決策\(k=j\)入隊並維護隊尾嚴格單調遞減性,將\(<\max(0,j-c_iw_i)\)的決策出隊即可。

這樣時間復雜度等於狀態數,為\(\mathrm O(nm)\),因為單調隊列是均攤\(\mathrm O(1)\)的。空間上呢,\(dp\)數組可以用滾動數組滾掉第一維,於是空間復雜度為\(\mathrm O(n+m)\)

下面貼代碼:

int q[M],head,tail;//單調隊列 
int dp[2][M+1]; 

memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)//考慮到第i個物品 
	for(int j=0;j<w[i];j++){//分組考慮 
		head=tail=0;//清空隊列 
		for(int k=j;k<=m;k+=w[i]){//遍歷此組中的所有狀態 
			while(head<tail&&q[head]<k-c[i]*w[i])head++;//出隊 
			while(head<tail&&w[i]*dp[i-1&1][q[tail-1]]-q[tail-1]*v[i]<=w[i]*dp[i-1&1][k]-k*v[i])tail--;//維護隊尾單調性 
			q[tail++]=k;//入隊 
			dp[i&1][k]=(w[i]*dp[i-1&1][q[head]]-q[head]*v[i]+k*v[i])/w[i];//此時隊首是最優決策 
		}
	}
//目標為dp[n&1][m]

\(2\)兩種解法的比較

復雜度上,不管是時間還是空間,都是解法\(2\)更優。不過單調隊列相對難推、難寫,要是在比賽中,不卡時間不卡常的話,還是寫解法\(1\)為妙。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM