题目:
判断给定的一个数组,是不是可以分成两个数组之和相等的数组
Example:
[1,2,6,3]可以分成[1,2,3]和[6]
解析:
此题可以用0,1背包问题来解决,分成的两个数组之和,一定为整个数组之和的一半,所以将背包容量设为初始数组之和的一半即可,最后在判断背包所装的容量是不是整个数组之和的一半。
代码:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
int n;
cin >> n;
vector<int> a(n);
int sum = 0;
for (int i = 0; i < n; ++i)
{
cin >> a[i];
sum += a[i];
}
vector<vector<int> > dp(n + 1, vector<int>(sum / 2 + 1, 0));
for (int i = 1; i <= n; ++i)
{
for (int j = 1; j <= sum / 2; ++j)
{
if (j >= a[i - 1])
dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - a[i - 1]] + a[i - 1]);
else
dp[i][j] = dp[i - 1][j];
}
}
bool flag = (sum - 2 * dp[n][sum / 2] == 0);
if (flag)
cout << "YES" << endl;
else
cout << "NO" << endl;
return 0;
}