POJ 1935 树形DP

比较好的一道树形DP，与POJ 2486 对比可以加深对树形DP的理解。

题意：给你一棵树和它的根节点，然后给出一些需要遍历的点，问你遍历这些点至少需要多少时间。

解析：dp【i】【0】表示从i节点出发遍历需要的点然后回到i节点，dp【i】【1】则表示不回来，状态想出来了，但是转移的时候并不像之前所做的树形DP一样，要求对于每个子节点都转移（因为有些子树中并没有要求遍历的点），这个怎么处理呢?在DFS的过程中，用plan记录下这棵子树中有没有目标节点，有的话则进行转移，没有则continue；

贴代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn = 50005;
struct edge
{
    int u,v,w,next;
}e[maxn << 1];
int head[maxn],tot,dp[maxn][2],n,m,s;
bool plan[maxn];
void add_edge(int u,int v,int w);
void dfs(int f,int u);
int main()
{
    while(scanf("%d%d",&n,&s) != EOF)
    {
        memset(head,-1,sizeof(head));
        memset(plan,false,sizeof(plan));
        tot = 0;
        for(int i = 1;i < n;i ++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);
            add_edge(v,u,w);
        }
        scanf("%d",&m);
        for(int i = 1;i <= m;i ++)
        {
            int a;
            scanf("%d",&a);
            plan[a] = true;
        }
        memset(dp,0,sizeof(dp));
        dfs(s,s);
      //  for(int i = 1;i <= n;i ++) printf("I:%d  plan:%d\n",i,plan[i]);
        printf("%d\n",dp[s][1]);
    }
    return 0;
}
void add_edge(int u,int v,int w)
{
    e[tot].u = u,e[tot].v = v,e[tot].w = w;
    e[tot].next = head[u],head[u] = tot ++;
}
void dfs(int f,int u)
{
    for(int i = head[u];i != -1;i = e[i].next)
    {
        int v = e[i].v;
        if(v == f) continue;
        dfs(u,v);
        plan[u] = (plan[u] | plan[v]);
        if(plan[v])
        {
            dp[u][1] = min(dp[u][1] + dp[v][0] + e[i].w * 2,dp[u][0] + dp[v][1] + e[i].w);
            dp[u][0] = dp[u][0] + dp[v][0] + e[i].w * 2;
        }
    }
}