POJ 1935 트리 DP

비교적 좋은 나무 DP는 POJ2486과 비교하면 나무 DP에 대한 이해를 깊이 있게 할 수 있다.
제목: 나무 한 그루와 그 뿌리의 노드를 주고 그 뿌리를 옮겨다니는 데 적어도 얼마나 걸리는지 물어보자.
해석: dp[i][0]는 i노드에서 출발하여 필요한 점을 두루 훑어보고 i노드로 돌아간다는 뜻이고, dp[i][1]는 돌아오지 않고 상태가 생각났다는 뜻이지만 옮길 때 이전에 만든 트리 DP처럼 각 하위 노드를 옮기라고 요구하는 것은 아니다. (하위 나무에 두루 훑어보라고 하지 않은 점이 있기 때문이다) 이걸 어떻게 처리하지?DFS 과정에서 플랜으로 이 하위 트리에 목표 노드가 있는지 기록하고 어떤 것은 이동하고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
",i,plan[i]);
        printf("%d
",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;
        }
    }
}