树形DP POJ3659 好难的一道题

链接：http://poj.org/problem?id=3659

题意：给你n个点，给出n-1无向边表示这两个点相邻，问最小选几个点可以把整个区域覆盖（一个点被选会覆盖它相连的节点）

思路：可以随机把一个点看做根节点成为一个树，然后做树上dp

1我们定义:

dp[i][0]:自己被选，且所有子节点被覆盖的最小值

dp[i][1]:自己不被选，被父亲节点覆盖，且所有子节点被覆盖的最小值；

dp[i][2]:自己不被选，不被父亲节点覆盖，且所有子节点被覆盖的最小值

2初始化:

   dp[i][0]=1; dp[i][1]=0;  如果i是叶子节点dp[i][2]=inf 否则dp[i][2]=0; 

3设i的任意一个儿子的标号为u，有转移方程

      (1).如果当前节点被选,那么子节点可以取任何情况，累计每个子节点三中情况的最小值

       dp[i][0]+=min(dp[u][0],dp[u][1],dp[u][2] ); 

      (2).如果当前节点被父亲节点覆盖且没被选，那么子节点可以是自己被选，或者不被选不被父亲覆盖的最小值。

       dp[i][1]+=min(dp[u][0],dp[u][2] );

      (3).如果当前节点没有被儿子覆盖且没被选，和上一种情况基本相似，

           注意：必须有一个儿子被选，如果没被选当前节点就没有被覆盖!!! 

          所以我们在取最小值的时候要记录一下是否取到儿子被选的情况，同事维护一个最小值（dp[u][0]-dp[2]）;

         如果没有取到儿子被选的情况 那么dp[i][2]+=min;需要补上一个最小值，使最小值的那个子节点被选！！！

         具体的操作见代码，上面还有一些细节！！！

#include <iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
using namespace std;
const int maxn=10100;
const int inf=999999;
struct node
{
    int to,next;
}edge[maxn<<1];
int head[maxn];
int cnt=0;
void add(int u,int v)
{
    edge[cnt].to=v; edge[cnt].next=head[u]; head[u]=cnt++;
}
int n;
int dp[maxn<<1][3];
//dp[i][0] 自己被选
//dp[i][1] 自己没被选，父亲被选包含了他
//dp[i][2] 自己没被选，父亲没选没包含他他,儿子包含了他
bool vis[maxn<<1];

void dfs(int root)
{
    vis[root]=1;
    dp[root][0]=1;
    dp[root][1]=0;
    dp[root][2]=0;
    if(head[root]==-1)
    {
       dp[root][2]=inf;
       return;
    }
    int m=inf;
    bool flag=1;
    for(int i=head[root];i!=-1 ;i=edge[i].next)
    {
        int v=edge[i].to;
        if(!vis[v]) //这一块wa了一次，判断只要是儿子就加，但会重复算！！！
        {
            dfs(v);
            dp[root][0]+=min(dp[v][0],min(dp[v][1],dp[v][2]));
            dp[root][1]+=min(dp[v][0],dp[v][2]);
            if(dp[v][0]<=dp[v][2])
            {
                dp[root][2]+=dp[v][0];
                flag=0;
            }
            else
            {
                dp[root][2]+=dp[v][2];
                m=min(m,dp[v][0]-dp[v][2]);
            }
        }
    }
    if(flag)  dp[root][2]+=m;
}
int main()
{
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
    cnt=0;
    for(int i=1;i<n;++i)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b); add(b,a);
    }
    dfs(1);
    printf("%d\n",min(dp[1][0],dp[1][2]));
    return 0;
}