Я знаю, что уже поздно отвечать, но я думаю, что нашел простое решение этой проблемы.Мы перечисляем подмножества S
в лексикографическом порядке с использованием обратного отслеживания и проверяем sum
подмножества, сгенерированного до сих пор.
Когда sum
превышает k
, возникает интересная часть: нам нужно проверить,сгенерированный subset
является правильным подмножеством ранее сообщенных элементов.
Одним из решений является сохранение всех зарегистрированных подмножеств и проверка на включение, но это расточительно.
Вместо этого мы вычисляем разницумежду k
и sum
.Если в S
есть элемент e
, такой что e not in subset
и e <= (k - sum)
, то сгенерированный нами набор является правильным подмножеством ранее сообщенного подмножества, и мы можем спокойно его пропустить.
Вот полная рабочая программа на старом простом C ++, демонстрирующая идею:
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
typedef std::set<int> Set;
typedef std::vector<int> SubSet;
bool seen_before(const Set &universe, const SubSet &subset, int diff) {
Set::const_iterator i = std::mismatch(universe.begin(), universe.end(),
subset.begin()).first;
return i != universe.end() && *i <= diff;
}
void process(const SubSet &subset) {
if (subset.empty()) {
std::cout << "{}\n";
return;
}
std::cout << "{" << subset.front();
for (SubSet::const_iterator i = subset.begin() + 1, e = subset.end();
i != e; ++i) {
std::cout << ", " << *i;
}
std::cout << "}\n";
}
void generate_max_subsets_rec(const Set &universe, SubSet &subset,
long sum, long k) {
Set::const_iterator i = subset.empty()
? universe.begin()
: universe.upper_bound(subset.back()),
e = universe.end();
if (i == e) {
if (!seen_before(universe, subset, k - sum))
process(subset);
return;
}
for (; i != e; ++i) {
long new_sum = sum + *i;
if (new_sum > k) {
if (!seen_before(universe, subset, int(k - sum)))
process(subset);
return;
} else {
subset.push_back(*i);
if (new_sum == k)
process(subset);
else
generate_max_subsets_rec(universe, subset, new_sum, k);
subset.pop_back();
}
}
}
void generate_max_subsets(const Set &universe, long k) {
SubSet subset;
subset.reserve(universe.size());
generate_max_subsets_rec(universe, subset, 0, k);
}
int main() {
int items[] = {1, 2, 3, 4, 5};
Set u(items, items + (sizeof items / sizeof items[0]));
generate_max_subsets(u, 7);
return 0;
}
Вывод всех максимальных подмножеств в лексикографическом порядке, по одному на строку:
{1, 2, 3}
{1, 2, 4}
{1, 5}
{2, 5}
{3, 4}