0-1揹包問題(動態規劃)

關於01揹包問題的優化
一、01揹包問題介紹

揹包問題是經典的動態規劃問題之一;

常見的01揹包問題就是說有一堆物品,現在要裝入一個容器中,這些物品的重量和價值各不一致,而容器的重量又是有限的,沒種物品只能裝1個或者不裝(0個),求當重量限定爲w時,這些物品能裝進去組合成的最高價值是多少?

分析:我們首先將物品排成一排(隨機),依次標記爲1號,2號。。。。然後從一號開始依次往裏放,放的時候判斷當前物品是不是應該放進去:

如果當前物品放進去之後的最高總價值 比 不放進去的最高總價值 大 ,那麼就是要放入,然後總價值取放入之後的

反之不放入,總價值依然取之前的值。

並且,在對同一個物品判斷時,從0依次增加容器的容量,直到上限w

當前物品放進去之後的最高總價值 = 上一號物品判斷時(容量 = 當前容量 - 當前物品重量 時)總價值 + 當前物品的價值

不放進去的最高總價值 = 上一號物品判斷的時候(容量 = 當前容量 時)總價值

所以有僞代碼如下:

    if (當前物品重量 > 當前容量) {
        此時最高總價值 = 當前物品不放進去的最高總價值;  // 此時物品重量比容量大,放不進去,只能取放不進去的情況
    } else {
        此時最高總價值 = max(當前物品不放進去的最高總價值, 當前物品放進去的最高總價值);
    }

因此,我們可以用一個狀態表來對整個過程進行描述。

假設現在是有三個物品 (不是三種)一號、二號、三號,重量分別爲3,2,5; 價值分別爲7,4,8; 當前容器容量爲8,求最大價值。

首先,第一行,爲沒有任何物品的時候,全部置0;

第二行,判斷一號物品能不能放:

當容量擴充爲3的時候,一號物品終於能放,價值爲7,由於當前也只有一號,所以後面都是7;

第三行,判斷二號物品能不能放:

當容量擴充爲2的時候,二號物品終於能放,價值爲4,與不放進去的最高價值(一號物品在此容量時的值:0)進行比較,所以成功放進去;

當容量擴充爲3的時候,二號物品一定放時,價值爲4,與不放進去的最高價值(一號物品在此容量時的值:7)進行比較,決定不放進去了;

。。。

當容量擴充爲5的時候,二號物品一定放時,最高價值爲(一號物品在(當前容量 - 二號物品的重量)時的價值+二號物品的價值)

=(一號物品在容量爲(5-2)的時候 + 二號物品的價值)= 7 + 4 =11

。。。【如下表】

容量 0 1 2 3 4 5 6 7 8
無 0 0 0 0 0 0 0 0 0
一號 0 0 0 7 7 7 7 7 7
二號 0 0 4 7 7 11 11 11 11
三號 0 0 4 7 7 11 11 12 15
所以有代碼如下:

public static int getMaxValue(int[] weight, int[] value, int w) {
    int n = weight.length;
    int[][] table = new int[n + 1][w + 1];
    for (int i = 1; i <= n; i++) {     // 物品遍歷(第0行肯定全是0,所以不用遍歷)
        for (int j = 0; j <= w; j++) { // 揹包大小遞增
            if (weight[i-1] > j) {     // 當前物品i的重量比揹包容量j大,裝不下,只能是不裝
                table[i][j] = table[i - 1][j];
            } else {                   // 裝得下,Max{不裝物品i, 裝物品i}
                table[i][j] = Math.max(table[i - 1][j], table[i - 1][j - weight[i-1]] + value[i-1]);
            }
        }
    }
    return table[n][w];
}

提問:爲什麼要【n+1】行?

——因爲之後每一個數都需要與上一行做比較,一號物品放入的時候也需要與沒放入的時候做比較,所以空出一行作爲“沒有物品的時候”;

爲什麼要【w+1】列?

——因爲在容量遞增時,第一個物品能放的時候,需要與容量減去當前物品的重量,也就是容量爲0的時候做比較,所以,也是需要一列作爲“0容量的時候”;

爲什麼是weight[i - 1] 和 value[i - 1]?

——因爲傳入的重量和價值必然是從一號物品開始,而我們的表是從“沒有物品”和“0容量”開始,多了一行和一列,所以 i 是當前物品的標號,而當前物品下標爲【標號-1】;

二、空間複雜度優化

很顯然,上面算法的空間複雜度爲矩陣大小【n+1】*【w+1】。

n可能不大,但是實際應用傷的w可能會很大,這樣造成了比較大的空間佔用。

而仔細觀察發現,矩陣中的值只與當前值的左上角的矩陣裏的值有關,如圖二號物品容量爲4時的價值7,只與綠色標註值有關。

容量 0 1 2 3 4 5 6 7 8
無 0 0 0 0 0 0 0 0 0
一號 0 0 0 7 7 7 7 7 7
二號 0 0 4 7 7 11 11 11 11
三號 0 0 4 7 7 11 11 12 15
  並且我們是按行進行更新的,所以我們用一個一維數據就能進行狀態的更替,不過要注意更新的方向。

所以依賴關係爲:下面依賴上面,右邊依賴左邊;

如果正常地從左到右邊,那麼右邊面等待更新的值需要的依賴(左邊)就會被覆蓋掉,所以應該每行從右邊開始更新。代碼如下:

public static int getMaxValueByOne(int[] weight, int[] value, int w) {
    int n = weight.length;
    int[] table = new int[w + 1];

    for (int i = 1; i <= n; i++) {
        for (int j = w; j >= 0; j--) {
            if (weight[i-1] <= j) {
                table[j] = Math.max(table[j], table[j - weight[i-1]] + value[i-1]);
            }
        }
    }
    return table[w];
}

如此一來,01揹包的空間複雜度就降爲了O(w)。

清風吹斜陽(轉載http://www.cnblogs.com/Xieyang-blog/p/9432711.html)
——唯有更努力才能讓一切都是最好的安排。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章