题目链接:点击打开链接
题意:
给6种等级的物品,每种物品有 ni 个;
求是否能平分为两份,其等级之和相等;
当然单独一个物品是不能分成两半的;
理解:
最开始想了下,直接暴力的一个01背包;
就是把这些物品全部按等级值存在一个数组中;
之后用01背包计算;
即:一些数之和一定是小于等于所有的和的一半才能满足两份接近相等的关系;
递推式为:dp[i] = max(dp[i], dp[i - v[i]] + v[i])
初始值为:dp[0 ~ sum / 2] = 0;
最终求的就是:dp[sum / 2];
但是,这样是超时的;
因为sum / 2 <= 10000, 而 v.size() <= 20000;
所以直接超时了;
于是又看到了神的答复,对这些数进行二进制分解;
何为二进制分解呢?
解说是这样的:
对于每一种物品,其和 s 是固定的,只是我们要将其划成几堆,而不是每一个一堆;
这样的堆组合起来又可以表示 [1, s] 之间的每一个数;
而二进制分解就是,把和 s 分解成二进制数;
即:s = 1 + 2 + 4 + ... + 2 ^ (k - 1) + s - 2 ^ k;
这些数为何可以表示 [1, s] 之间的每一个数呢?
且看这个博客
这样弄了过后就不会超时了;
其实这是一个多重背包问题;
代码如下:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <map>
#include <set>
#include <queue>
#include <stack>
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int MIN_INF = 1e-7;
const int MAX_INF = (1e9) + 7;
#define X first
#define Y second
int main() {
int n[7];
int t = 1;
while (cin >> n[1] >> n[2] >> n[3] >> n[4] >> n[5] >> n[6] &&
(n[1] || n[2] || n[3] || n[4] || n[5] || n[6])) {
cout << "Collection #" << t++ << ":" << endl;
vector<int> v;
int sum = 0;
for (int j = 1; j < 7; ++j) {
sum += n[j] * j;
for (int i = 1; i <= n[j]; i <<= 1) {
v.push_back(j * i);
n[j] -= i;
}
if (n[j] >= 0) {
v.push_back(n[j] * j);
}
}
if (sum % 2) {
cout << "Can't be divided." << endl;
cout << endl; // PE
continue;
}
vector<int> dp(sum / 2 + 10, 0);
for (int i = 0; i < v.size(); ++i) {
for (int j = sum / 2; j >= v[i]; --j) {
dp[j] = max(dp[j], dp[j - v[i]] + v[i]);
}
}
if (dp[sum / 2] != sum / 2) {
cout << "Can't be divided." << endl;
cout << endl;
}
else {
cout << "Can be divided." << endl;
cout << endl;
}
}
return 0;
}