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

HDU-2586-裸LCA入门-tarjan离线

时间:2018-12-06 20:44:22      阅读:167      评论:0      收藏:0      [点我收藏+]

标签:const   ext   scan   iostream   ios   std   target   pair   ace   

http://acm.hdu.edu.cn/showproblem.php?pid=2586

给出一颗树和边权,询问两点距离。

    考虑tarjan离线做法,做法很巧妙,当前进行到u,对他的儿子v,当v子树tarjan完成之后把v合并到u上。当遍历完所有v之后,对与u有关的询问进行查找,若第二个询问点v被访问过,那么lca(u,v)就是v目前被合并到的根上。还有记录d[u]表示根到u的距离。

  最后答案就是d[u]+d[v]-2*d[lca(u,v)]。

 

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<map>
 5 #include<set>
 6 #include<stack>
 7 #include<deque>
 8 #include<bitset>
 9 #include<unordered_map>
10 #include<unordered_set>
11 #include<queue>
12 #include<cstdlib>
13 #include<ctype.h>
14 #include<ctime>
15 #include<functional>
16 #include<algorithm>
17 #include<bits/stdc++.h>
18 using namespace std;
19 #define LL long long 
20 #define pii pair<int,int>
21 #define mp make_pair
22 #define pb push_back
23 #define fi first
24 #define se second
25 #define inf 0x3f3f3f3f
26 #define debug puts("debug")
27 #define mid ((L+R)>>1)
28 #define lc (id<<1)
29 #define rc (id<<1|1)
30 const int maxn=40010;
31 const int maxm=50050;
32 const double PI=acos(-1.0);
33 const double eps=1e-6;
34 const LL mod=1e9+7;
35 LL gcd(LL a,LL b){return b==0?a:gcd(b,a%b);}
36 LL lcm(LL a,LL b){return a/gcd(a,b)*b;}
37 LL qpow(LL a,LL b,LL c){LL r=1; for(;b;b>>=1,a=a*a%c)if(b&1)r=r*a%c;return r;}
38 template<class T>
39 void prt(T v){for(auto x:v)cout<<x<< ;cout<<endl;}
40 struct Edge{int u,v,w,next;};
41 
42 vector<pii>g[maxn],q[maxn];
43 int d[maxn],f[maxn],qu[220],qv[220],ans[220];
44 bool vis[maxn];
45 int getf(int u){return f[u]==u?u:f[u]=getf(f[u]);}
46 void tarjan(int u){
47     vis[u]=1;
48     for(pii e:g[u]){
49         int v=e.fi,w=e.se;
50         if(!vis[v]){
51             d[v]=d[u]+w;
52             tarjan(v);
53             f[v]=u;
54         }
55     }
56     for(pii e:q[u]){
57         int v=e.fi,id=e.se;
58         if(vis[v]){
59             ans[id]=getf(v);
60         }
61     }
62 }
63 int main(){
64     int t,n,m,i,j,u,v,w;
65     scanf("%d",&t);
66     while(t--){
67         scanf("%d%d",&n,&m);
68         for(i=1;i<=n;++i){
69             g[i].clear();
70             q[i].clear();
71             f[i]=i;
72             vis[i]=0;
73         }
74         for(i=1;i<n;++i){
75             scanf("%d%d%d",&u,&v,&w);
76             g[u].pb(mp(v,w));
77             g[v].pb(mp(u,w));
78         }
79         for(i=1;i<=m;++i){
80             scanf("%d%d",qu+i,qv+i);
81             q[qu[i]].pb(mp(qv[i],i));
82             q[qv[i]].pb(mp(qu[i],i));
83         }
84         tarjan(1);
85         for(i=1;i<=m;++i){
86             printf("%d\n",d[qu[i]]+d[qv[i]]-2*d[ans[i]]);
87         }
88     }
89     return 0;
90 }

 

HDU-2586-裸LCA入门-tarjan离线

标签:const   ext   scan   iostream   ios   std   target   pair   ace   

原文地址:https://www.cnblogs.com/zzqc/p/10078343.html

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