leetcode416. Partition Equal Subset Sum

Caicloud 發(fā)布于2019-08-16 15:10 / 1814人閱讀

摘要：題目要求假設(shè)有一個(gè)全為正整數(shù)的非空數(shù)組，將其中的數(shù)字分為兩部分，確保兩部分?jǐn)?shù)字的和相等。而這里的問題等價(jià)于，有個(gè)物品，每個(gè)物品承重為，問如何挑選物品，使得背包的承重搞好為所有物品重量和的一般。

題目要求

Given a non-empty array containing only positive integers, find if the array can be partitioned into two subsets such that the sum of elements in both subsets is equal.

Note:
1.Each of the array element will not exceed 100.
2.The array size will not exceed 200.
Example 1:

Input: [1, 5, 11, 5]

Output: true

Explanation: The array can be partitioned as [1, 5, 5] and [11].
Example 2:

Input: [1, 2, 3, 5]

Output: false

Explanation: The array cannot be partitioned into equal sum subsets.

假設(shè)有一個(gè)全為正整數(shù)的非空數(shù)組，將其中的數(shù)字分為兩部分，確保兩部分?jǐn)?shù)字的和相等。

思路和代碼

這和0-1背包問題是完全一樣的，01背包問題是指假設(shè)有n個(gè)物品，每個(gè)物品中為weight[i]，假設(shè)背包的承重為k，問如何選擇物品使得背包中的承重最大。而這里的問題等價(jià)于，有n個(gè)物品，每個(gè)物品承重為input[i]，問如何挑選物品，使得背包的承重搞好為所有物品重量和的一般。

假設(shè)我們已經(jīng)知道前i個(gè)物品是否能夠構(gòu)成重量k，我們令該值為dpi，那么第i+1個(gè)物品是否能夠構(gòu)成重量取決于dpi和dpi], 即i+1個(gè)物品能夠構(gòu)成重量k有兩種情況，第一種選擇了i+1個(gè)物品，則要尋找前i個(gè)物品是否構(gòu)成k-input[i+1]的重量，第二種就是沒有選擇第i+1個(gè)物品，則直接判斷前i個(gè)物品是否已經(jīng)構(gòu)成了k的重量。

代碼如下：

    public boolean canParition(int[] nums) {
        int sum = 0;
        for(int n : nums) {
            sum += n;
        }
        if(sum % 2 != 0) return false;
        int half = sum / 2;
        boolean[][] sums = new boolean[nums.length+1][half+1];
        for(int i = 0 ; i<=nums.length ; i++) {
            for(int j = 0 ; j<=half ; j++) {
                if(i==0 && j==0){
                    sums[i][j] = true;
                }else if(i==0) {
                    sums[i][j] = false;
                }else if(j==0){
                    sums[i][j] = true;
                }else {
                    sums[i][j] = sums[i-1][j] || (nums[i-1] <= j ? sums[i-1][j-nums[i-1]] : false);
                }
            }
        }
        return sums[nums.length][half];
    }