洛谷P2016 战略游戏 - 树形DP

一、题目

战略游戏

二、分析

dp1[i] : 第i个节点站士兵，照亮以i为节点的子树所需最少的士兵数；

dp0[i] : 第i个节点不站士兵，照亮以i为节点的子树所需最少的士兵数；

状态转移方程：

三、代码

#include<bits/stdc++.h>
using namespace std;
const int N=2000;

int h[N],e[N*2],ne[N*2],idx;
int n;
int dp0[N],dp1[N];
bool st[N];

void add(int a,int b)
{
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}


void dfs(int root)
{
    dp1[root]=1;
    dp0[root]=0;
    for(int i=h[root];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(!st[j])
        {
            st[j]=true;
            dfs(j);
            st[j]=false;
            dp1[root]+=min(dp1[j],dp0[j]);
            dp0[root]+=dp1[j];
        }
    }
}


int main()
{
    cin>>n;
    for(int i=0;i<n;i++)
        h[i]=-1;
    for(int i=1;i<=n;i++)
    {
        int k,a,b;
        cin>>a>>k;
        for(int j=1;j<=k;j++)
        {
            cin>>b;
            add(a,b),add(b,a);
        }
    }

    int Min=0x3f3f3f3f;
    for(int i=0;i<n;i++)
    {
        st[i]=true;
        dfs(i);
        st[i]=false;
        Min=min(Min,min(dp0[i],dp1[i]));
    }

    cout<<Min<<endl;
    return 0;
}