JZOJ 1075. 【GDKOI2006】新红黑树
\(\text{Problem}\)
A君和B君在玩一种叫做新红黑树的游戏,即在一棵由红枝和黑枝构成的树上轮流砍树枝,每次砍一枝,A君每次只能砍红枝,B君每次只能砍黑枝,当其中某人已经没有树枝砍的时候,由另外一人砍,直到砍完全部树枝。树枝是带权的,每个人的总分是他砍的树枝的权值之和,那些由于其他树枝被砍掉而与根失去联系的树枝会自动消失。每次由A君先砍,设“D=A君的得分-B君的得分”,A君想让D最大,而B君想让D最小,A君和B君都是极其聪明的人,他们始终以最优策略进行整个游戏,你知道最后的D值是多少吗?
\(1\le n\le 20\)
\(\text{Solution}\)
考虑两个 \(dfs\) 互相暴搜,二进制记录边信息
然后记忆化
以前没打过,今天考场一遍过了?!
比较慢,其实可以预处理删边后同时影响的边的二进制信息
这样就不需要每次删边后再一条一条边的删去子树影响
\(\text{Code}\)
#include <cstdio>
#include <cstring>
#include <vector>
#define RE register
using namespace std;
const int N = 25, INF = 2e9;
int n, f[1<<21][2], dep[N];
struct edge{int u, v, c, w;}e[N];
vector<int> g[N];
void prepare(int x, int fa)
{
dep[x] = dep[fa] + 1;
for(RE int i = 0; i < n - 1; i++)
{
if (e[i].u == x && e[i].v ^ fa)
{
prepare(e[i].v, x), g[x].push_back(i);
for(RE int j = 0; j < g[e[i].v].size(); j++) g[x].push_back(g[e[i].v][j]);
}
else if (e[i].v == x && e[i].u ^ fa)
{
prepare(e[i].u, x), g[x].push_back(i);
for(RE int j = 0; j < g[e[i].u].size(); j++) g[x].push_back(g[e[i].u][j]);
}
}
}
int dfs1(int s);
int dfs2(int s);
int dfs1(int s)
{
if (f[s][0] != -INF) return f[s][0];
if (s == (1 << n - 1) - 1) return 0;
int ret = -INF, bz = 0;
for(RE int i = 0; i < n - 1; i++)
{
if (((s >> i) & 1) || (e[i].c != 1)) continue;
int ss = (s | (1 << i)), x = e[i].u, y = e[i].v;
if (dep[x] < dep[y]) swap(x, y);
for(RE int j = 0; j < g[x].size(); j++) ss |= (1 << g[x][j]);
ret = max(ret, e[i].w - dfs2(ss)), bz = 1;
}
if (!bz) return -dfs2(s);
return f[s][0] = ret;
}
int dfs2(int s)
{
if (f[s][1] != -INF) return f[s][1];
if (s == (1 << n - 1) - 1) return 0;
int ret = -INF, bz = 0;
for(RE int i = 0; i < n - 1; i++)
{
if (((s >> i) & 1) || (e[i].c != -1)) continue;
int ss = (s | (1 << i)), x = e[i].u, y = e[i].v;
if (dep[x] < dep[y]) swap(x, y);
for(RE int j = 0; j < g[x].size(); j++) ss |= (1 << g[x][j]);
ret = max(ret, e[i].w - dfs1(ss)), bz = 1;
}
if (!bz) return -dfs1(s);
return f[s][1] = ret;
}
int main()
{
scanf("%d", &n), ++n;
for(RE int i = 0; i < n - 1; i++)
scanf("%d%d%d%d", &e[i].u, &e[i].v, &e[i].c, &e[i].w), ++e[i].u, ++e[i].v;
prepare(1, 0);
for(RE int i = 0; i < (1 << n - 1); i++) f[i][0] = f[i][1] = -INF;
printf("%d\n", dfs1(0));
}