代码随想录算法训练营第42天| ● 01背包问题，你该了解这些！ ● 01背包问题，你该了解这些！ 滚动数组 ● 416. 分割等和子集

416. 分割等和子集

已解答
中等
相关标签
相关企业
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。

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

提示：

• 1 <= nums.length <= 200
• 1 <= nums[i] <= 100

代码

package __dp

func canPartition(nums []int) bool {
   /*
      计算总和
      背包一半bag[n-1][sum/2]
   */
   n := len(nums)
   sum := 0
   for i := 0; i < n; i++ {
      sum += nums[i]//计算总和
   }
   if sum%2 == 1 {
      return false
   }
   sum /= 2
   bag := make([]int, sum+1)
   for i := 0; i < n; i++ {//填背包
      for j := sum; j >= nums[i]; j-- {
         bag[j] = max(bag[j], bag[j-nums[i]]+nums[i])
      }
   }
   if bag[sum] == sum {
      return true
   }
   return false
}