HDU 2586 How far away ? << LCA+tarjin求树上任两点最短距离裸题

题意

给你一棵树和每条边的权值,m次询问,求任两点间距离,LCA+tarjin裸题。

思路

能用的算法大概也很多吧,但是就为了练一手LCA+tarjin我就来做这题了,够裸了,还能验板子233。

这个算法主要是用到了并查集,找LCA,在找到询问对的时候存下LCA。然后最后利用公式结合tarjin得来的距离和LCA直接得出答案。

所以说此处我们进行遍历的目的有两个——1.找到目标点对的LCA,2.找到所有点距离初始点的距离。最后的答案是通过以上我们得到的信息算出来的。

该题详情请见代码。

代码

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int N=4e4+5;
 4 typedef pair<int,int> PII;
 5 struct edge{
 6     int to,len;
 7 };
 8 struct askedge{
 9     int from,to,lca;
10 }ae[300];//询问对
11 int n,m;
12 vector<edge> G[N];//邻接表
13 vector<PII> pp[N];//存储每个点相对的询问点
14 int dist[N];//距离
15 int fa[N];//并查集
16 int Find(int x)
17 {
18     if(x==fa[x]) return x;
19     else return fa[x]=Find(fa[x]);
20 }
21 void Union(int x,int y)
22 {
23     x=Find(x);y=Find(y);
24     if(x!=y) fa[y]=x;
25 }
26 void dfs(int now)
27 {
28     fa[now]=now;//初始化,分离成init()函数也没问题
29     for(int i=0;i<pp[now].size();i++)//看当前这个点有没有它的点对已经被搜索到
30     {
31         PII &v=pp[now][i];
32         if(dist[v.first]!=-1){
33             ae[v.second].lca=fa[v.first];//如果存在,他们的lca就是当前点对象的爸爸
34         }
35     }
36     for(int i=0;i<G[now].size();i++)//遍历搜索
37     {
38         edge &v=G[now][i];
39         if(dist[v.to]==-1)
40         {
41             dist[v.to]=dist[now]+v.len;
42             dfs(v.to);
43             Union(now,v.to);//回溯时合并
44         }
45     }
46 }
47 int main()
48 {
49     int T;
50     scanf("%d",&T);
51     while(T--)
52     {
53         scanf("%d%d",&n,&m);
54         for(int i=0;i<=n;i++)//初始化
55         {
56             pp[i].clear();
57             G[i].clear();
58         }
59         for(int i=0;i<=m;i++)
60             pp[i].clear();
61         for(int i=0,u,v,len;i<n-1;i++)//读入图
62         {
63             scanf("%d%d%d",&u,&v,&len);
64             G[u].push_back(edge {v,len});
65             G[v].push_back(edge {u,len});
66         }
67         for(int i=1;i<=m;i++)//读入询问
68         {
69             int u,v;
70             scanf("%d%d",&u,&v);
71             pp[u].push_back(make_pair(v,i));
72             pp[v].push_back(make_pair(u,i));
73             ae[i]={u,v,-1};
74         }
75         memset(dist,-1,sizeof(dist));
76         dist[1]=0;
77         dfs(1);//开始tarjin
78         for(int i=1;i<=m;i++)//对每次询问输出答案
79         {
80             askedge &u=ae[i];
81             printf("%d\n",dist[u.from]+dist[u.to]-2*dist[u.lca]);//公式
82         }
83     }
84     return 0;
85 }

 

posted @ 2018-09-10 19:46  computer_luo  阅读(266)  评论(0编辑  收藏  举报