动态规划——分割等和子集

分割等和子集

416. 分割等和子集 - 力扣（LeetCode）

• 给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。

  示例 1：

  输入：nums = [1,5,11,5]
  输出：true
  解释：数组可以分割成 [1, 5, 5] 和 [11] 。
  

  示例 2：

  输入：nums = [1,2,3,5]
  输出：false
  解释：数组不能分割成两个元素和相等的子集。
  

  提示：

  • 1 <= nums.length <= 200
  • 1 <= nums[i] <= 100

• 暴力递归，每一个数拿或者不拿两种情况，求出对应的和，如果最后和为数组的和的一半则true

class Solution {
public:
    int max(int a, int b) {
        int asw;
        asw = a>b? a : b;
        return asw;
    }
    bool canPartition(vector<int>& nums) {
        int sum = 0;
        for(auto i : nums) {
            sum += i;
        }
        if(sum % 2 != 0) return false;

        if(DFS(nums,0,sum/2) == sum/2) return true;
        else return false;
    }

    int DFS(vector<int>& nums, int index, int restSum) {
        if(restSum < 0) {
            return -1;
        }
        if(index == nums.size()) {
            return 0;
        }
        int take = 0;
        int t = DFS(nums, index+1, restSum - nums[index]);
        if(t != -1) {
            take = nums[index] + t;
        }
        int notake = DFS(nums, index+1, restSum);
        return max(take,notake);
    }
};

• 发现存在重复计算的子集，使用记忆化搜索

class Solution {
public:
    vector<vector<int>> dp;
    int max(int a, int b) {
        int asw;
        asw = a>b? a : b;
        return asw;
    }
    bool canPartition(vector<int>& nums) {
        int sum = 0;
        for(auto i : nums) {
            sum += i;
        }
        if(sum % 2 != 0) return false;

        dp.resize(nums.size()+1, vector<int>(sum/2+1, -1));
        if(DFS(nums,0,sum/2) == sum/2) return true;
        else return false;
    }

    int DFS(vector<int>& nums, int index, int restSum) {
        if(restSum < 0) {
            return -1;
        }
        if(dp[index][restSum] != -1) return dp[index][restSum];
        if(index == nums.size()) {
            return dp[index][restSum] = 0;
        }
        int take = 0;
        int t = DFS(nums, index+1, restSum - nums[index]);
        if(t != -1) {
            take = nums[index] + t;
        }
        int notake = DFS(nums, index+1, restSum);
        return dp[index][restSum] = max(take,notake);
    }
};

• 直接通过记忆化搜索的表格，推导动态规划。

class Solution {
public:
    int max(int a, int b) {
        int asw;
        asw = a>b? a : b;
        return asw;
    }
    bool canPartition(vector<int>& nums) {
        int sum = 0;
        for(auto i : nums) {
            sum += i;
        }
        if(sum % 2 != 0) return false;
        int n = nums.size();

        vector<vector<int>> dp(n+1, vector<int>(sum/2+1, 0));
        for(int i = n-1; i >= 0; --i) {
            for(int j = 0; j < sum/2+1; ++j) {
                if(j < nums[i]) {
                    dp[i][j] = dp[i+1][j];
                }
                else {
                    dp[i][j] = max(dp[i + 1][j], nums[i] + dp[i + 1][j - nums[i]]);
                }
            }
        }
        if(dp[0][sum/2] == sum/2) return true;
        else return false;
    }
};