LeetCode hot 100—分割等和子集

题目

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

示例

示例 1:

复制代码
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。

示例 2:

复制代码
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。

分析

这是一个典型的 0 - 1 背包问题的变种。要判断是否能将数组 nums 分割成两个子集,使得两个子集的元素和相等,等价于判断数组中是否存在一个子集,其元素和等于数组元素总和的一半。

动态规划

代码解释

计算数组总和 :使用 std::accumulate 函数计算数组 nums 的总和 totalSum

判断总和是否为偶数 :如果总和是奇数,那么无法将数组分割成两个和相等的子集,直接返回 false

确定目标和 :如果总和是偶数,目标和 targettotalSum / 2

初始化动态规划数组dp[i][j] 表示前 i 个元素中是否能选出和为 j 的子集。初始化 dp[i][0]true,表示和为 0 的子集总是可以找到(不选任何元素)。

动态规划填充 dp 数组

  • 如果当前元素 nums[i - 1] 大于目标和 j,则不能选择该元素,dp[i][j] = dp[i - 1][j]
  • 否则,可以选择不选当前元素或者选当前元素,dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]]

返回结果 :最终结果为 dp[n][target],表示前 n 个元素中是否能选出和为 target 的子集。

时间复杂度:O(), 是数组长度, 是数组元素总和的一半

空间复杂度:O()

cpp 复制代码
class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int totalSum = std::accumulate(nums.begin(), nums.end(), 0);
        // 如果总和是奇数,无法分割成两个和相等的子集
        if (totalSum % 2 != 0) {
            return false;
        }
        int target = totalSum / 2;
        int n = nums.size();
        // dp[i][j] 表示前 i 个元素中是否能选出和为 j 的子集
        std::vector<std::vector<bool>> dp(n + 1, std::vector<bool>(target + 1, false));
        // 初始化:和为 0 的子集总是可以找到(不选任何元素)
        for (int i = 0; i <= n; ++i) {
            dp[i][0] = true;
        }
        // 动态规划填充 dp 数组
        for (int i = 1; i <= n; ++i) {
            for (int j = 1; j <= target; ++j) {
                // 如果当前元素大于目标和 j,则不能选择该元素
                if (nums[i - 1] > j) {
                    dp[i][j] = dp[i - 1][j];
                } else {
                    // 否则,可以选择不选当前元素或者选当前元素
                    dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]];
                }
            }
        }
        return dp[n][target];
    }
};    
相关推荐
YouQian77226 分钟前
Traffic Lights set的使用
算法
go54631584652 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法
QQ_4376643142 小时前
C++11 右值引用 Lambda 表达式
java·开发语言·c++
aramae2 小时前
大话数据结构之<队列>
c语言·开发语言·数据结构·算法
大锦终2 小时前
【算法】前缀和经典例题
算法·leetcode
想变成树袋熊3 小时前
【自用】NLP算法面经(6)
人工智能·算法·自然语言处理
cccc来财3 小时前
Java实现大根堆与小根堆详解
数据结构·算法·leetcode
liulilittle3 小时前
C++/CLI与标准C++的语法差异(一)
开发语言·c++·.net·cli·clr·托管·原生
小狄同学呀3 小时前
VS插件报错,g++却完美编译?API调用错因分析
c++
程序员编程指南4 小时前
Qt 数据库连接池实现与管理
c语言·数据库·c++·qt·oracle