假设我们有一个仅包含正数的非空数组,我们必须确定该数组是否可以划分为两个子集,以使两个子集中的元素之和相同。因此,如果输入类似于[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]
让我们看下面的实现以更好地理解-
#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