原题链接:https://vjudge.net/problem/UVA-1627
分类:0-1背包
备注:二分图,背包变体
按照紫书的思路,将不互相认识的连在一起,这样形成连通图必须要能够变为二分图,否则不合题意,然后黑白染色,两种颜色对应两个分组,互换一下分组,组员数量的差值就会变化,可以据此来进行DP。
开始以为,只要对每个连通图看两种情况即可,每次都求出最优结果。WA了之后再看别人的解法,仔细一想发现直接这样的状态转移是一种贪心的做法。在洛谷上有一句总结:对付这种“01背包”,常见的状态设定是:用 dp[i][j] 表示前 i 个物品是否能凑出 j 的价值。根据这种思路,就才能对全局最优都有把握。
还是挺难的一个题,像是二分图这里估计自己都想不到。
#include <bits/stdc++.h>
using namespace std;
const int N = 100 + 5;
int t, n, tot, co[N], g[N][N], diff[N], dp[N][N * 2];
vector<int> e[N][2];
bool dfs(int u, int o) {
co[u] = o;
e[tot][o].push_back(u);
for (int v = 1; v <= n; v++) {
if (u == v || (g[u][v] && g[v][u])) continue;
if (co[v] != -1 && co[v] == co[u]) return false;
if (co[v] == -1 && !dfs(v, o ^ 1)) return false;
}
return true;
}
bool init() {
memset(co, -1, sizeof(co));
tot = 0;
for (int i = 1; i <= n; i++) {
if (co[i] != -1) continue;
e[tot][0].clear();
e[tot][1].clear();
if (!dfs(i, 0)) return false;
diff[tot++] = e[tot][0].size() - e[tot][1].size();
}
return true;
}
int main(void) {
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
scanf("%d", &t);
for (int kase = 1; kase <= t; kase++) {
if (kase != 1) puts("");
scanf("%d", &n);
memset(g, 0, sizeof(g));
for (int i = 1; i <= n; i++) {
int x;
while (~scanf("%d", &x) && x != 0) {
g[i][x] = 1;
}
}
if (!init()) puts("No solution");
else {
memset(dp, 0, sizeof(dp));
dp[0][0 + n] = 1;
for (int i = 0; i < tot; i++) {
for (int j = -n; j <= n; j++) {
if (dp[i][j + n]) {
dp[i + 1][j + n + diff[i]] = 1;
dp[i + 1][j + n - diff[i]] = 1;
}
}
}
int ans;
for (ans = 0; ans <= n; ans++) {
if (dp[tot][n + ans]) break;
if (dp[tot][n - ans]) {
ans *= -1; break;
}
}
vector<int> team[2];
for (int i = tot - 1, o; i >= 0; i--) {
if (dp[i][ans - diff[i] + n]) {
o = 0; ans -= diff[i];
} else {
o = 1; ans += diff[i];
}
for (int j = 0; j < e[i][o].size(); j++)
team[0].push_back(e[i][o][j]);
for (int j = 0; j < e[i][o ^ 1].size(); j++)
team[1].push_back(e[i][o ^ 1][j]);
}
for (int i = 0; i < 2; i++) {
printf("%d", team[i].size());
for (int j = 0; j < team[i].size(); j++)
printf(" %d", team[i][j]);
puts("");
}
}
}
return 0;
}