Acwing 1077. 皇宫看守（树形DP）

 

根据题目，我们先来分析每个点可能拥有的状态，由题目可知，当一个点上有士兵时，士兵可以看到所有与该点直接相连的点，所以一个点如果可以被看到，可以被分成以下几种状态

（1）、该点可以被它的父节点看见。

（2）、该点可以被自己看见。

（3）、该点可以被它的子节点看见。

故可以设置状态f[u][0],f[u][1],f[u][2]分别表示这三种状态。

状态转移：

（1）、如果该点可以被它的父节点看见（该结点不放士兵），那么该点的孩子结点既可以被自己看看见，也可以被孩子结点的子节点看见，从二者中取一个最小值。

（2）、如果该点可以被它的孩子结点看到（该节点不放士兵），那么应该枚举该点会被哪些孩子结点看到，加上所有其他孩子结点的选择（可以自己看到自己，也可以由孩子结点看到），枚举所有情况取一个最小值。

（3）、如果该点可以被自己看见，那么，该点的孩子结点既可以被父节点，孩子结点，自己看见，从这三种情况中取一个最小值即可。

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=1510;
int h[N],e[N],ne[N],idx,w[N];
int n,m,f[N][3];
bool st[N];
void add(int u,int v)
{
    e[idx]=v;
    ne[idx]=h[u];
    h[u]=idx++;
}
void dfs(int u)
{
    f[u][2]=w[u];
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        dfs(j);
        f[u][0]+=min(f[j][1],f[j][2]);
        f[u][2]+=min(f[j][0],min(f[j][1],f[j][2]));
    }
    f[u][1]=1e9;
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        //f[u][0]代表所有min(f[son][1],f[son][2])的和
        f[u][1]=min(f[u][1],f[j][2]+f[u][0]-min(f[j][1],f[j][2]));
    }
}
int main()
{
    cin>>n;
    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++)
    {
        int pos;
        cin>>pos;
        cin>>w[pos];
        cin>>m;
        for(int j=1;j<=m;j++)
        {
           int v;
           cin>>v;
           add(pos,v);
           st[v]=true;
        }
    }
    int root=1;
    while(st[root])
    root++;
    dfs(root);
    cout<<min(f[root][1],f[root][2])<<endl;
    return 0;
}