「Usaco2008 Jan」电话网络

一道比较神的树形DP，恕我太菜，一开始想的是三维的。。
实际上并不需要三维，只需要用0,1,2表示即可
$dp[u][0]$表示当前点不放，他的某一个儿子覆盖他的代价
$dp[u][1]$表示当前点放，他的子树也被覆盖的代价
$dp[u][2]$表示覆盖当前点的子树除当前点之外的代价

思考一下为什么会有$dp[u][2]$，因为一个点不选，他的儿子也可以不选
只要他的儿子的子树全部合格就行…
代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
const int INF = 2147483647;
const int maxn = 10007;
int dp[maxn][3];
int n;
 
struct node{
    int to,next;
}edge[maxn*2];

int cnt,head[maxn];
void add(int from,int to){
    edge[++cnt].to=to;
    edge[cnt].next=head[from];
    head[from]=cnt;
}
 
void dfs(int u,int fa){
    dp[u][0]=dp[u][2]=0;
    dp[u][1]=1;
    int son=0,flag=0,delta=INF;
    for(int i=head[u];i;i=edge[i].next){
        int to=edge[i].to;
        son++;
        if(to==fa)continue;
        dfs(to,u);
        if(dp[to][1]<dp[to][0])flag=1;else delta=min(delta,dp[to][1]-dp[to][0]);
        dp[u][0]+=min(dp[to][1],dp[to][0]);
        dp[u][1]+=min(min(dp[to][0],dp[to][1]),dp[to][2]);
        dp[u][2]+=dp[to][0];
    }
    if(flag==0)dp[u][0]+=delta;
    if(son==1&&edge[head[u]].to==fa){
        dp[u][0]=n;
        dp[u][1]=1;
        dp[u][2]=0;
    }
}
 
 
 
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    cout<<min(dp[1][1],dp[1][0]);
    return 0;
}