划分和相等的子集-LintCode

最新推荐文章于 2024-08-25 16:00:00 发布

原创最新推荐文章于 2024-08-25 16:00:00 发布 · 3.3k 阅读

3 ·

CC 4.0 BY-SA版权

LintCode 专栏收录该内容

428 篇文章

订阅专栏

本文探讨了如何通过动态规划解决子集划分问题，即给定一个包含正整数的数组，判断其是否能被分为两个元素和相等的子集。文章提供了具体的实现思路和代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给一只含有正整数的非空数组, 找到这个数组是否可以划分为两个元素和相等的子集。

注意事项：
所有数组元素不超过100.
数组大小不超过200.

样例：
给一数组 [1, 5, 11, 5] , 返回 true ,
两个子集:[1, 5, 5], [11]
给一数组 [1, 2, 3, 9] , 返回 false

思路：
动态规划，对于数组nums，判断奇偶性，若为奇数，肯定不可能分成两个相等的数，
若为偶数，令sum为数组nums元素和的一半。构建数组dp[len][sum],dp[i][j]表示nums的第一个元素到
第i个元素构成的集合中，是否存在使得和为j的子集。
状态转换方程为：

d p [i] [j] = {d p [i - 1] [j] | | d p [i - 1] [j - n u m s [i - 1]], d p [i - 1] [j], j > = n u m s [i - 1] o t h e r w i s e

$dp[i][j]=\left\{\begin{array}{cc} dp[i-1][j]||dp[i-1][j-nums[i-1]], & j>=nums[i-1]\\ dp[i-1][j], & otherwise \end{array}\right.$

#ifndef C588_H
#define C588_H
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
    /*
    * @param nums: a non-empty array only positive integers
    * @return: true if can partition or false
    */
    bool canPartition(vector<int> &nums) {
        // write your code here
        int sum = 0;
        for (auto c : nums)
            sum += c;
        if (sum % 2 != 0)
            return false;
        else
            sum /= 2;
        int len = nums.size();
        vector<vector<bool>> dp(len + 1, vector<bool>(sum + 1, false));
        for (int i = 0; i <= len; ++i)
            dp[i][0] = true;
        for (int i = 1; i <= len; ++i)
        {
            for (int j = 1; j <= sum; ++j)
            {
                if (j >= nums[i - 1])
                    dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]];
                else
                    dp[i][j] = dp[i - 1][j];
            }
        }
        return dp[len][sum];
    }
};
#endif