1. 程式人生 > >01揹包,完全揹包,多重揹包的個人總結

01揹包,完全揹包,多重揹包的個人總結

大一剛接觸揹包問題的時候就覺得繞。那時候真的是一點程式碼基礎都沒有強行去理解。每次都是以失敗告終,一直到大二都還不會寫揹包問題。

後來某次模擬賽之後碰到了揹包問題,覺得這個還是挺簡單的,終於是下定決心準備搞一搞這個東西了。

有了一定的基礎理解起來就比以前容易多了。

首先,先分清楚這三個揹包問題。

1.01揹包:有n種物品與承重為m的揹包。每種物品只有一件,每個物品都有對應的重量weight[i]與價值value[i],求解如何裝包使得價值最大。

2.完全揹包:有n種物品與承重為m的揹包。每種物品有無限多件,每個物品都有對應的重量weight[i]與價值value[i],求解如何裝包使得價值最大。

3.多重揹包:有n種物品與承重為m的揹包。每種物品有有限件num[i],每個物品都有對應的重量weight[i]與價值value[i],求解如何裝包使得價值最大。

啊,說真的大一的我聽了這些已經趴下去睡覺了。

但是真的比對起來會發現,其實這些問題都是很類似的,三種揹包就是三個約束條件不一樣而已。

那麼針對不同的約束條件,開始解決問題。

(一)關於01揹包

吶,為什麼叫它01揹包呢,因為裝進去就是1,不裝進去就是0.所以針對每個物品就兩種狀態,裝,不裝(請允許我用這麼老套的開篇,相信聽過很多次揹包講解的人,大多都是這個開篇的)所以咯,我這個揹包啊,只要有足夠大的空間,這個物品是有可能被裝進去的咯。

所以有狀態轉移方程

dp[i][j] = max( dp[i-1][j] , dp[i-1][ j - weight[i] ] + value[i] )

然後二維陣列的程式碼寫法分分鐘就出來了,反正都是跟前一個狀態去轉移,也沒有什麼寫法上的限制。

#include<bits/stdc++.h>
using namespace std;
int dp[1005][1005];
int weight[1005];
int value[1005];
int main()
{
    int n,m;
    cin>>m>>n;
    memset(dp,0,sizeof(dp));//陣列清空,其實同時就把邊界給做了清理
    for(int i=1; i<=n; i++)
        cin>>weight[i]>>value[i];
    //從1開始有講究的因為涉及到dp[i-1][j],從0開始會越界
    for(int i=1; i<=n; i++)//判斷每個物品能否放進
    {
        for(int j=0; j<=m; j++)//對每個狀態進行判斷
        //這邊兩重for都可以倒著寫,只是需要處理最邊界的情況,滾動陣列不一樣
        {
            if(j>=weight[i])//能放進
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-weight[i]]+value[i]);

            else dp[i][j]=dp[i-1][j];//不能放進
        }
    }
    cout<<dp[n][m]<<endl;
    return 0;
}

然後啊,我們來仔細分析分析就會發現,這個陣列開銷還是很大的,因為是二維的,萬一哪個資料一大,分分鐘記憶體超限,因此有了下邊的解法

傳說中的---------------滾動陣列!!!

啊?什麼是滾動陣列。

說白了二維陣列只是把每個物品都跑一遍,然後到最後一個物品的時候輸出答案,那麼過程值只是計算的時候用一次,我沒必要存下來。所以用一個數組去滾動儲存,然後用後一個狀態的值去覆蓋前面一個狀態。然後形象的叫它:滾動陣列(ma!dan!一點都不形象,我理解了好久)

好吧,假裝很形象。

那麼問題來了,怎麼樣用一維的去代替二維的工作,或者說怎麼去思考。這是一個難點。

那麼我們想,遍歷物品的那個for肯定不能省去,然後裡邊的for也不能省。。。。那麼。就把那個i給他刪了吧,好像確實沒啥用哦。

然後就出現了這樣的程式碼

 for(int i=1; i<=n; i++)
    {
        for(int j=weight[i]; j<=m; j++)
        {
            dp[j]=max(dp[j],dp[j-weight[i]]+value[i]);
        }
    }

看上去好像很厲害的樣子,但是這個絕對是錯的,因為第二個for裡存在某個dp[j]被改動過,然後再次影響到更大的j。就像我們再對一個數組進行移位操作,一不小心就全部成了一樣的數。(別笑,你們以前肯定碰到過|||- -)

額。。回到正題,上邊的程式碼會有重複影響,確實歪打正著的碰上了另一個揹包。這個另說,現在附上正確的思路。

#include<bits/stdc++.h>
using namespace std;
int dp[1005];//滾動陣列的寫法,省下空間省不去時間
int weight[1005];
int value[1005];
int main()
{
    int n,m;
    cin>>m>>n;
    memset(dp,0,sizeof(dp));
    for(int i=1; i<=n; i++)
        cin>>weight[i]>>value[i];
    for(int i=1; i<=n; i++)//對每個數判斷,可反
    {
        for(int j=m; j>=weight[i]; j--)//這裡這個迴圈定死,不能反,反了就是完全揹包
        {
            dp[j]=max(dp[j],dp[j-weight[i]]+value[i]);//其實不斷在判斷最優解,一層一層的
        }
    }
    cout<<dp[m]<<endl;
    return 0;
}
其實就是規定從m開始迴圈,保證了選擇這個物品時,肯定不會重複使用狀態。

(二)關於完全揹包

就像先前講的,完全揹包是每個物品都無限,那麼我只要對著一個性價比最高的物品狂選就是了啊。??

是嗎?有瑕疵啊!

反例一批一批的啊,認死了選價效比最高的,不一定是完全填滿揹包的啊,萬一最後一個是剛好填滿揹包的,而且價格湊起來剛好比全選價效比最高的物品高的情況比比皆是啊。

啊?什麼,特判最後一個狀態?

你在搞笑嗎|||- -,那我再往前推到倒數第二件,第三件咋辦。總不能對每個物品都特判吧。

所以正解就是動態規劃。狀態轉移方程如下:

dp[i][j] = max ( dp[i-1][j - k*weight[i]] +k*value[i] )     0<=k*weight[i]<=m

這樣看是不是還要多一重for去算k(既放入這個物品的個數)

那麼這裡二維陣列就不如一維的了。

上程式碼:

#include<bits/stdc++.h>
using namespace std;
int dp[100005];//m
struct Node{
    int a,b;
}node[1005];//n

int main(){
    int n;
    while(~scanf("%d",&n)){
        for(int i=0;i<n;i++){
            scanf("%d%d",&node[i].a,&node[i].b);
        }
        int m;
        scanf("%d",&m);
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++){
            for(int j=node[i].b;j<=m;j++){//這樣就是完全揹包
                dp[j]=max(dp[j],dp[j-node[i].b]+node[i].a);
            }
        }
        printf("%d\n",dp[m]);
    }
    return 0;
}

特意換了一種寫法讓整體的程式碼結構看上去更像揹包一點。

其中第二個for中是從小到大遍歷的。這樣就是要利用這種影響(因為物品不知道取幾個呀,能取就減咯,這種影響正好就是這個題目的意思)

(三)關於多重揹包

理解了前面兩種揹包,那麼第三種揹包理解起來就毫不費力了

首先這種可以把物品拆開,把相同的num[i]件物品 看成 價值跟重量相同的num[i]件不同的物品,那麼!!是不是就轉化成了一個規模稍微大一點的01揹包了。對不對!!對不對!!

No BB, show me the code!!

哦, 客官,您要的程式碼:

#include<bits/stdc++.h>
using namespace std;
int dp[1005];
int weight[1005],value[1005],num[1005];
int main()
{
    int n,m;
    cin>>n>>m;
    memset(dp,0,sizeof(dp));
    for(int i=1; i<=n; i++)
        cin>>weight[i]>>value[i]>>num[i];
        
    for(int i=1; i<=n; i++)//每種物品
        
        for(int k=0; k<num[i]; k++)//其實就是把這類物品展開,呼叫num[i]次01揹包程式碼
        
            for(int j=m; j>=weight[i]; j--)//正常的01揹包程式碼
            
                dp[j]=max(dp[j],dp[j-weight[i]]+value[i]);
    
    cout<<dp[m]<<endl;
    return 0;
}

那只是一種理解方法,其實正規的應該是這樣的

dp[i][j] = max ( dp[i-1][j - k*weight[i]] +k*value[i] )     0<=k<=num[i](這個跟完全揹包差點就一毛一樣了啊喂|||- -)

那麼還是用滾動陣列來寫,而且還又優化了下

程式碼參考:http://blog.csdn.net/acdreamers/article/details/8563283

#include<bits/stdc++.h>
using namespace std;
const int N = 1005;

int dp[N];
int c[N],w[N],num[N];
int n,m;

void ZeroOne_Pack(int cost,int weight,int n)//吧01揹包封裝成函式
{
    for(int i=n; i>=cost; i--)
        dp[i] = max(dp[i],dp[i-cost] + weight);
}

void Complete_Pack(int cost,int weight,int n)//把完全揹包封裝成函式
{
    for(int i=cost; i<=n; i++)
        dp[i] = max(dp[i],dp[i-cost] + weight);
}

int Multi_Pack(int c[],int w[],int num[],int n,int m)//多重揹包
{
    memset(dp,0,sizeof(dp));
    for(int i=1; i<=n; i++)//遍歷每種物品
    {
        if(num[i]*c[i] > m)
            Complete_Pack(c[i],w[i],m);
            //如果全裝進去已經超了重量,相當於這個物品就是無限的
            //因為是取不光的。那麼就用完全揹包去套
        else
        {
            int k = 1;
            //取得光的話,去遍歷每種取法
            //這裡用到是二進位制思想,降低了複雜度
            //為什麼呢,因為他取的1,2,4,8...與餘數個該物品,打包成一個大型的該物品
            //這樣足夠湊出了從0-k個該物品取法
            //把複雜度從k變成了logk
            //如k=11,則有1,2,4,4,足夠湊出0-11個該物品的取法
            while(k < num[i])
            {
                ZeroOne_Pack(k*c[i],k*w[i],m);
                num[i] -= k;
                k <<= 1;
            }
            ZeroOne_Pack(num[i]*c[i],num[i]*w[i],m);
        }
    }
    return dp[m];
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>m>>n;
        for(int i=1; i<=n; i++)
            cin>>c[i]>>w[i]>>num[i];
        cout<<Multi_Pack(c,w,num,n,m)<<endl;
    }
    return 0;
}

幫助個人學習用,如果幫到大家的話也是極好的。