leetcode 416 分割等和子集

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

注意:

每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:

输入: [1, 5, 11, 5]

输出: true

解释: 数组可以分割成 [1, 5, 5] 和 [11].
 

示例 2:

输入: [1, 2, 3, 5]

输出: false

解释: 数组不能分割成两个元素和相等的子集.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-equal-subset-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

记忆化搜索

代码：

class Solution {
public:
    int f[200][20001] = {0};
    bool dfs(vector<int> &nums,int k,int sum) {
        if(sum < 0 || k >= 200) return false;
        if(f[k][sum]) return f[k][sum] - 1;
        for(int i = k;i < nums.size();i ++) {
            if(dfs(nums,i + 1,sum - nums[i])) return f[k][sum] = 2;
        }
        return (f[k][sum] = 1) - 1;
    }
    bool canPartition(vector<int>& nums) {
        int sum = 0;
        for(int i = 0;i < nums.size();i ++) {
            sum += nums[i];
            f[i][0] = 2;
        }
        if(sum % 2) return false;
        sum /= 2;
        for(int i = 0;i < nums.size();i ++) {
            if(dfs(nums,i + 1,sum - nums[i])) return true;
        }
        return false;
    }
};