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>
 5using namespace std;
 6# define max(a,b) ((a)>(b)?(a):(b))
 7int dp[50001][2];
 8bool need[50001];
 9int g[50001],nxt[100005],val[100005],v[100005],c=0;
10inline 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}

17void 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}

30int 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

posted on 2010-11-16 02:06 yzhw 阅读(143) 评论(0)  编辑 收藏 引用 所属分类: DPgraph


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


<2011年2月>
303112345
6789101112
13141516171819
20212223242526
272812345
6789101112

导航

统计

公告

统计系统

留言簿(1)

随笔分类(227)

文章分类(2)

OJ

最新随笔

搜索

积分与排名

最新评论

阅读排行榜