C ++中的分区相等子集总和
假设我们有一个仅包含正数的非空数组,我们必须确定该数组是否可以划分为两个子集,以使两个子集中的元素之和相同。因此,如果输入类似于[1,5,11,5],则输出为true。由于此数组可以划分为[1、5、5]和[11]
为了解决这个问题,我们将遵循以下步骤-
n:=数组的大小
和:=0
对于我:=0至n–1
sum:=sum+nums[i]
如果总和是奇数,则返回false
总和:=总和/2
创建一个称为dp的数组,大小为sum+1
dp[0]:=true
对于i,范围为0至n–1
dp[j]:=dp[j]或dp[j-x],不为0
x:=nums[i]
对于j:=求和到j–x
返回dp[sum]
例子(C++)
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h> using namespace std; class Solution { public: bool canPartition(vector<int>& nums) { int n = nums.size(); int sum = 0; for(int i =0;i<n;i++)sum+=nums[i]; if(sum&1)return false; sum/=2; vector <bool> dp(sum+1); dp[0] = true; for(int i =0;i<n;i++){ int x = nums[i]; for(int j =sum;j-x>=0;j--){ dp[j]=dp[j] || dp[j-x]; } } return dp[sum]; } }; main(){ Solution ob; vector<int> v = {1,5,11,5}; cout << ob.canPartition(v); }
输入值
[1,5,11,5]
输出结果
1