本文共 791 字,大约阅读时间需要 2 分钟。
题目:
判断给定的一个数组,是不是可以分成两个数组之和相等的数组
Example:
[1,2,6,3]可以分成[1,2,3]和[6]
解析:
此题可以用0,1背包问题来解决,分成的两个数组之和,一定为整个数组之和的一半,所以将背包容量设为初始数组之和的一半即可,最后在判断背包所装的容量是不是整个数组之和的一半。
代码:
#include#include #include using namespace std;int main(){ int n; cin >> n; vector a(n); int sum = 0; for (int i = 0; i < n; ++i) { cin >> a[i]; sum += a[i]; } vector > dp(n + 1, vector (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;}
转载地址:http://avwab.baihongyu.com/