代码随想录算法训练营第四十二天| 416. 分割等和子集

LeetCode 416. 分割等和子集

链接：416. 分割等和子集

思路：

这是典型的01背包问题，即可以选取的物品每种只有一个。题目要求在一个数组里找到两个子集，使得两个子集的元素和相等，也就是说每个子集的元素和为数组元素总和的一半。首先遍历数组计算出整个数组的总和，如果总和不能被2整除，则说明无论如何都不可能分割成两个等和子集。然后初始化dp数组，背包大小为子集元素和的大小，即为总和的一半。

接下来定义下标：dp[i]表示大小为i的背包最多能用数组nums里的元素填满的大小是多少。01背包的递归公式为：dp[j] = max(dp[j], dp[j - nums[i]] + nums[i])。这个公式可以解释成，当背包大小为j的时候，检测看背包空出nums[i]的大小后所能放的东西大小dp[j - nums[i]]，再加上nums[i]的大小是否大于不放nums[i]。因为采用的是滚动数组，遍历顺序需要先遍历物品再遍历背包，因为背包一定要倒序遍历，否则会把同一个物品选用多次（那就不是01背包而是完全背包了）。

最后检查dp[target]是否为总和的一半，如果不为一半说明无法分割成两个和相等的子集，反之亦然。

代码：

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = 0;
        for (int i = 0; i < nums.size(); i++)
            sum += nums[i];
        // 无法分成两份
        if (sum % 2 != 0)
            return false;
        int target = sum / 2;
        // 01背包，大小为总和的一半，初始化为0
        vector<int>dp(target+1, 0);
        // 先遍历物品，再遍历背包大小
        for (int i = 0; i < nums.size(); i++)
        {
            // 从大到小遍历
            for (int j = target; j >= nums[i]; j--)
                // 01背包递推公式
                dp[j] = max(dp[j], dp[j - nums[i]] + nums[i]);
        }

        if (dp[target] != target)
            return false;
        else
            return true;
    }
};

总结：

本题为01背包应用题，首先要分清楚什么时候用01背包什么时候用完全背包，题目中很显然给出了每个元素只能用一次，所以一定是01背包。其次要注意遍历顺序，滚动数组的时候背包大小一定要倒序遍历，否则会重复采用之前的物品，其次倒序遍历的时候不能放在外面遍历，必须要先遍历物品再遍历背包。