pku 1935 Journey 树形DP

pku 1935 Journey 树形DP

简明题意：给出一个城市的道路网(是一棵树)，每条路有一定的权值，一个人在第k点，给出一些城市列表，问这个人游览完这些城市最小花费为多少
解法：一条最优的路线肯定是这样

有且仅有一条路线是单向的。
下面定义状态:
dp[i][0]为游览完以i为根节点的子树（仅仅游览需要游览的城市，如果没有即为0）且最后回到i节点需要的最短长度
dp[i][1]为不需要回到i节点的最短长度
dp[i][0]=sum(dp[j][0]+val[i][j](如果dp[j][0]不为0或者j为需要访问的城市)),j为i的孩子节点
dp[i][1]=dp[i][0]-max(dp[j][0]-dp[j][1]+val[i][j](如果dp[j][0]不为0或者j为需要访问的城市))
最后结果就是dp[start][1]
程序如下

 1 # include  < cstdio >
 2 # include  < cstring >
 3 # include  < vector >
 4 // # include <algorithm>
 5 using   namespace  std;
 6 # define max(a,b) ((a) > (b) ? (a):(b))
 7 int  dp[ 50001 ][ 2 ];
 8 bool  need[ 50001 ];
 9 int  g[ 50001 ],nxt[ 100005 ],val[ 100005 ],v[ 100005 ],c = 0 ;
10 inline  void  insert( int  a, int  b, int  p)
11 {
12   v[c]=b;
13   val[c]=p;
14   nxt[c]=g[a];
15   g[a]=c++;
16}
17 void  solve( int  pos, int  pre)
18 {
19     int maxnum=0;
20     dp[pos][0]=dp[pos][1]=0;
21     for(int p=g[pos];p!=-1;p=nxt[p])
22       if(v[p]!=pre)
23       {
24          solve(v[p],pos);
25          maxnum=max(dp[v[p]][0]-dp[v[p]][1]+(dp[v[p]][0]||need[v[p]]?val[p]:0),maxnum);
26          dp[pos][0]+=dp[v[p]][0]+(dp[v[p]][0]||need[v[p]]?2*val[p]:0);
27       }
28     dp[pos][1]=dp[pos][0]-maxnum;
29}
30 int  main()
31 {
32    int n,start,num;
33    scanf("%d%d",&n,&start);
34    memset(g,-1,sizeof(g));
35    memset(need,false,sizeof(need));
36    memset(dp,-1,sizeof(dp));
37    for(int i=1;i<n;i++)
38    {
39       int a,b,p;
40       scanf("%d%d%d",&a,&b,&p);
41       insert(a,b,p);
42       insert(b,a,p);
43    }
44    scanf("%d",&num);
45    while(num--)
46    {
47      int t;
48      scanf("%d",&t);
49      need[t]=true;
50    }
51    solve(start,-1);
52    printf("%d\n",dp[start][1]);
53   // system("pause");
54    return 0;
55}
56
57