#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> readVector(int n) {
vector<int> vec(n);
for (int i = 0; i < n; ++i) {
scanf("%i", &vec[i]);
}
sort(vec.begin(), vec.end(), greater<int>());
return vec;
}
vector<int> items;
vector<int> bags;
vector<bool> set;
void genSet(int bagSize, int mask = 0, int i = 0) {
if (i >= items.size()) {
set[mask] = true;
return;
}
if (items[i] <= bagSize) {
genSet(bagSize - items[i], mask | (1 << i), i + 1);
}
genSet(bagSize, mask, i + 1);
}
int main() {
int n, m;
scanf("%i %i", &n, &m);
items = readVector(n);
bags = readVector(m);
const int spaceSize = 1 << n;
vector<bool> prevSet(spaceSize, false);
vector<bool> nextSet;
prevSet[0] = true;
for (int i = 0; i < bags.size(); ++i) {
set.clear();
set.resize(spaceSize, false);
genSet(bags[i]);
nextSet.clear();
nextSet.resize(spaceSize, false);
for (int u = 0; u < spaceSize; ++u) {
for (int v = 0; v < spaceSize; ++v) {
if (prevSet[u] && set[v]) {
nextSet[u | v] = true;
}
}
}
if (nextSet[spaceSize - 1]) {
printf("%i", i + 1);
return 0;
}
nextSet.swap(prevSet);
}
puts("NIE");
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 | #include <cstdio> #include <vector> #include <algorithm> using namespace std; vector<int> readVector(int n) { vector<int> vec(n); for (int i = 0; i < n; ++i) { scanf("%i", &vec[i]); } sort(vec.begin(), vec.end(), greater<int>()); return vec; } vector<int> items; vector<int> bags; vector<bool> set; void genSet(int bagSize, int mask = 0, int i = 0) { if (i >= items.size()) { set[mask] = true; return; } if (items[i] <= bagSize) { genSet(bagSize - items[i], mask | (1 << i), i + 1); } genSet(bagSize, mask, i + 1); } int main() { int n, m; scanf("%i %i", &n, &m); items = readVector(n); bags = readVector(m); const int spaceSize = 1 << n; vector<bool> prevSet(spaceSize, false); vector<bool> nextSet; prevSet[0] = true; for (int i = 0; i < bags.size(); ++i) { set.clear(); set.resize(spaceSize, false); genSet(bags[i]); nextSet.clear(); nextSet.resize(spaceSize, false); for (int u = 0; u < spaceSize; ++u) { for (int v = 0; v < spaceSize; ++v) { if (prevSet[u] && set[v]) { nextSet[u | v] = true; } } } if (nextSet[spaceSize - 1]) { printf("%i", i + 1); return 0; } nextSet.swap(prevSet); } puts("NIE"); return 0; } |
English