Leetcode刷题笔记 416. 分割等和子集

知识点：动态规划、背包问题
时间：2020年10月11日
题目链接：https://leetcode-cn.com/problems/partition-equal-subset-sum/

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

注意:

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

示例1
输入：
[1, 5, 11, 5]
输出：
true
解释:
数组可以分割成 [1, 5, 5] 和 [11].

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

思路

1. 转换成背包问题
2. 剪枝 总和不能整除的
3. 数据是否能放满半个背包

代码

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum_nums = 0;
        int max_num = 0;
        for(int i=0;i<nums.size();i++){
            sum_nums+=nums[i];
            if(max_num>nums[i])
                max_num=nums[i];
        }
        int target = sum_nums/2;
        if(sum_nums%2||target<max_num){return false;}
        vector<int> dp(target+1,0);
        dp[0] = 1;
        for(int i = 0;i < nums.size();i++){
            for(int j = target;j >=nums[i] ;j--){
                dp[j] = dp[j] | dp[j-nums[i]];
            }
        }
        return dp[target];
    }
};
int main()
{
    vector<int> nums{1,5,11,5};
    Solution s;
    cout<<s.canPartition(nums);
    return 0;
}

今天也是爱zz的一天哦！