양수만 포함하는 비어 있지 않은 배열이 있다고 가정하면 배열이 두 하위 집합의 요소 합계가 같도록 두 하위 집합으로 분할될 수 있는지 찾아야 합니다. 따라서 입력이 [1,5,11,5]와 같으면 출력이 true가 됩니다. 이 배열은 [1, 5, 5] 및 [11]
로 분할될 수 있으므로이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- n :=배열의 크기
- 합계 :=0
- i:=0 ~ n – 1
- 합계 :=합 + 숫자[i]
- 합계가 홀수이면 false를 반환
- 합계 :=합 / 2
- Sum + 1 크기의 dp라는 배열 하나 만들기
- dp[0] :=사실
- 0 ~ n – 1 범위의 i에 대해
- x :=숫자[i]
- for j :=j로 요약 – x
- dp[j] :=dp[j] 또는 dp[j - x](0이 아님)
- dp[합계]를 반환
예(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