leetcode 416. 分割等和子集

leetcode 416. 分割等和子集

题目详情

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

1. 每个数组中的元素不会超过 100
2. 数组的大小不会超过 200

• 示例 1:
  输入: [1, 5, 11, 5]
  输出: true
  解释: 数组可以分割成 [1, 5, 5] 和 [11].
• 示例 2:
  输入: [1, 2, 3, 5]
  输出: false
  解释: 数组不能分割成两个元素和相等的子集.

我的代码

// 没做出来。。。

执行用时为 0 ms 的范例

bool cmp(int &a, int &b)
{
    return a> b;
}
class Solution {
    vector<int> nums;
    int ave;
public:
    bool canPartition(vector<int>& nums) {
        if (nums.size() == 0) {
            return false;
        }
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) {
            sum += nums[i];
        }
        if (sum % 2 != 0) {
            return false;
        }
        sort(nums.begin(), nums.end(), cmp);
        int target = sum / 2;
        if (nums[nums.size() - 1] > target) {
            return false;
        }
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] > target) {
                return false;
            }
        }
        this->nums = nums;
        ave = target;
        int temp = 0;
        return Dfs(0, temp);
    }
    bool Dfs(int pos, int temp)
    {
        if (temp == ave) {
            return true;
        }
        for (int i = pos; i < nums.size(); i++) {
                if (temp + nums[i] > ave) {
                    return false;
                }
                if (Dfs(i + 1, temp + nums[i])) {
                    return true;
                }
        }
        return false;
    }
};

思考

参见官方解答