JZOJ 1075. 【GDKOI2006】新红黑树
阅读原文时间:2023年07月08日阅读:3

\(\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));
}