leetcode416——分割等和子集——java實現

題目要求:
在這裏插入圖片描述
分析:
這是一個0,1揹包問題。(更多揹包問題,請看這裏:傳送門:經典算法總結——揹包問題(java實現)【已完結】
這裏要注意,當sum爲奇數的時候,就直接不用做了,因爲sum/2是一個小數。

具體代碼如下:

class Solution {
    public boolean canPartition(int[] nums) {
        int len = nums.length;
        int sum = 0;
        for(int i = 0; i < len; i ++) {
            sum += nums[i];
        }
        boolean[][] dp = new boolean[len + 1][sum / 2 + 1];
        
        if(sum % 2 != 0) {
            return false;
        }
        
        //這裏也可以將初始條件設置爲dp[0][0] = 0,接下來等於多少計算機就會自己推算了
        for (int i = 1; i < sum / 2 + 1; i++) {
            if (nums[0] == i) {
                dp[0][i] = true;
            }
        }
        
        for(int i = 1; i < len; i ++) {
            for(int j = 0; j < sum / 2 + 1; j ++) {
                if(nums[i] > j) {
                    dp[i][j] = dp[i - 1][j];
                } else {
                    dp[i][j] = dp[i - 1][j - nums[i]] || dp[i - 1][j];
                }
            }
        }
        return dp[len - 1][sum / 2];
    }
}

上面的方法是看別人的,我覺得其實按照我的智商,是想不到直接new一個boolean的dp是怎麼去做的,所以我又寫了一個按照自己這種菜雞程度才能想到的代碼,具體如下:

class Solution {
    public boolean canPartition(int[] nums) {
        int len = nums.length;
        int sum = 0;
        for(int i = 0; i < len; i ++) {
            sum += nums[i];
        }
        if(sum % 2 != 0)
            return false;
        
        int target = sum / 2;
        int[][] dp = new int[len + 1][target + 1];
        
        for(int i = 0; i <= target; i ++) {
            dp[0][i] = 0;
        }
        
        for(int i = 1; i <= len; i ++) {
            for(int j = 0; j <= target; j ++) {
                if(j < nums[i - 1]) {
                    dp[i][j] = dp[i - 1][j];
                } else {
                    dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - nums[i-1]] + nums[i - 1]);
                }
            }
        }
        if(dp[len][target] == target) {
            return true;
        }
        return false;
    }
}
性能是不太好,但是勝在自己能想到。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章