码迷,mamicode.com
首页 > 其他好文 > 详细

小机房的树CODEVS 2370

时间:2014-12-27 18:53:08      阅读:160      评论:0      收藏:0      [点我收藏+]

标签:

小机房的树CODEVS 2370

————最近公共祖先和动态规划的完美结合

 【题目描述】

  小机房有棵焕狗种的树,树上有N个节点,节点标号为0到N-1,有两只虫子名叫飘狗和大吉狗,分居在两个不同的节点上。有一天,他们想爬到一个节点上去搞基,但是作为两只虫子,他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量(从父亲节点爬到此节点也相同),他们想找出一条花费精力最短的路,以使得搞基的时候精力旺盛,他们找到你要你设计一个程序来找到这条路,要求你告诉他们最少需要花费多少精力。

 【输入描述】

  第一行一个n,接下来n-1行每一行有三个整数u,v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。

  接下来m行每一行有两个整数 u ,v 表示两只虫子所在的节点

 【输出描述】

  一共有m行,每一行一个整数,表示对于该次询问所得出的最短距离

 【分析】

  求树上最短路,而且要求的是NlogN的算法,很容易能想到LCA,确实,LCA的确适合树上最短路

  Ps:LCA的具体内容博主的博客里有

 

  求出两点的LCA的过程中即可计算答案,我们设d(i,j)表示树上的i节点向上走2^j到达的节点所走的距离

  而d数组在初始化深度的时候即可计算

  时间复杂度O(NlogN)

  完美AC...

 【代码】

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <algorithm>
 5 using namespace std;
 6 
 7 const int MaxN=100001;
 8 
 9 struct list{
10     int to,next,w;
11 }e[MaxN];
12 
13 int head[MaxN],n,cnt=0;
14 int deep[MaxN],p[MaxN][22],d[MaxN][22];//d(i,j)表示i到第2^j祖先的距离 
15 int ans=0;
16 
17 void addedge(int u,int v,int w){
18     cnt++;
19     e[cnt].to=v;
20     e[cnt].w=w;
21     e[cnt].next=head[u];
22     head[u]=cnt;
23 }
24 
25 int lca(int u,int v){
26     if(deep[u]<deep[v])swap(u,v);
27     int t=deep[u]-deep[v],i,ans=0;
28     for(i=0;i<=21;i++)
29       if((1<<i)&t){
30         ans+=d[u][i];
31         u=p[u][i];
32       }
33     for(i=21;i>=0;i--)
34       if(p[u][i]!=p[v][i]){
35           ans+=d[u][i]+d[v][i];
36         u=p[u][i];
37           v=p[v][i];
38       }
39     if(u!=v)ans+=d[u][0]+d[v][0];
40     return ans;
41 }
42 
43 void dfs(int u){
44     int i;
45     for(i=1;i<=21;i++){
46       if(deep[u]<(1<<i))break;
47       p[u][i]=p[p[u][i-1]][i-1];
48       d[u][i]+=d[u][i-1]+d[p[u][i-1]][i-1];
49     }
50     for(i=head[u];i;i=e[i].next)
51         if(!deep[e[i].to]){
52             deep[e[i].to]=deep[u]+1;
53             d[e[i].to][0]=e[i].w;
54             p[e[i].to][0]=u;
55             dfs(e[i].to);
56         }
57 }
58 
59 int main(){
60   scanf("%d",&n);
61   int i,u,v,res,c;
62   for(i=1;i<n;i++){
63       scanf("%d%d%d",&u,&v,&c);
64       addedge(u,v,c);
65       addedge(v,u,c);
66   }
67   u=1;
68   for(i=0;i<n;i++)
69     if(!deep[i]){
70       p[i][0]=i;
71       deep[i]=1;
72       dfs(i);
73     }
74   scanf("%d",&c);
75   while(c--){
76       scanf("%d%d",&u,&v);
77     res=lca(u,v);
78     printf("%d\n",res);
79   }
80   return 0;
81 }

 

  无可否认,LCA的代码是有点长,但理解了还是很容易能敲出来的

 

  

小机房的树CODEVS 2370

标签:

原文地址:http://www.cnblogs.com/maopengsen/p/4188940.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!