皇宫看守 —— 树形DP + 状态机

题面

皇宫看守

思路

我们定义$f(i,j)$表示以节点$i$为根的子树，当前状态为$j$的所有方案，我们从中取最小的；

$j$的取值有三个，$0,1,2$；

$f(i,0)$ 表示点$i$的父节点有守卫(当前节点$i$不放)
$f(i,1)$ 表示点$i$的子节点有守卫(当前节点$i$不放)
$f(i,2)$ 表示点$i$有守卫

---

现在我们考虑状态转移；

因为$f(i,0)$是其父节点上有守卫，那么我们当前节点$i$就不需要守卫；
对于$i$的子节点来说，可以放守卫，也可以在自己的子节点上放守卫；

即$f(i,0)+=min(f(to,1),f(to,2))$

---

$f(i,2)$表示在当前点放守卫，那么其子节点三种状态都可以；

即$f(i,2)+=min(f(to,0),f(to,1),f(to,2))$

---

$f(i,1)$表示在子节点上放守卫(当前节点不放)；

因为一个点可能有多个子节点；

因此我们需要枚举一下，看看哪个子节点一定要放守卫的方案最优秀；

我们假设在子节点$v$上一定要放守卫，其他子节点可放可不放；

我们用$o$来表示除了$v$以外的所有子节点；

那么$f(i,1)=min(f(v,2)+min(f(o,1),f(o,2)))$

对于$min(f(o,1),f(o,2))$，我们发现这个方程跟$f(i,0)$类似；

因此我们可以统计全部的和;

再用这个和减去掉$min(f(v,1),f(v,2))$

Code

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N = 2e3 + 10;

//f(i,0) 表示点i的父节点有守卫
//f(i,1) 表示点i的子节点有守卫
//f(i,2) 表示点i有守卫
int f[N][3];

int head[N];

bool not_root[N];

struct Edge{
    int to,next;
}e[N];

int cnt,n,w[N];//w(i)是点权

void add(int u,int v){
    e[++cnt].to = v;
    e[cnt].next = head[u];
    head[u] = cnt;
}

void dfs(int u){
    f[u][2] = w[u];
    f[u][0] = 0;
    int sum = 0;
    for(int i=head[u];i;i=e[i].next){
        int to = e[i].to;
        dfs(to);
        f[u][0] += min(f[to][1],f[to][2]);
        sum += min(f[to][1],f[to][2]);
        f[u][2] += min(min(f[to][0],f[to][1]),f[to][2]);
    }
    for(int i=head[u];i;i=e[i].next){
        int to = e[i].to;
        //看看哪个节点固定放士兵比较好
        f[u][1] = min(f[u][1],f[to][2] + sum-min(f[to][1],f[to][2]));
    }
}

void solve(){
    memset(f,0x3f,sizeof f);
    int n;
    cin >> n;
    for(int i=1,u,num,v;i<=n;++i){
        cin >> u;
        cin >> w[u];
        cin >> num;
        while(num--){
            cin >> v;
            add(u,v);
            not_root[v] = 1;
        }
    }
    int root = 0;
    while(not_root[++root]) ;//找到根节点
    dfs(root);
    cout << min(f[root][1],f[root][2]) << '\n';
}

int main(){
    std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
    return 0;
}