Kruskal重构树+LCA || BZOJ 3732: Network

题面:https://www.lydsy.com/JudgeOnline/problem.php?id=3732

题解:Kruskal重构树板子

代码:

 

Kruskal重构树+LCA || BZOJ 3732: Network
 1 #include<cstdio>
 2 #include<algorithm>
 3 #include<vector>
 4 using namespace std;
 5 inline int rd(){
 6     int x=0,f=1; char c=getchar();
 7     while(c<'0'||c>'9'){if(c=='-')f=-1; c=getchar();}
 8     while(c>='0'&&c<='9'){x=x*10+c-'0'; c=getchar();}
 9     return f*x;
10 }
11 const int maxn=15005<<1,maxm=30005,maxk=20005,maxl=16;
12 int N,M,K,fa[maxn],n,D[maxn],A,B,F[maxn][maxl+2],Dep[maxn];
13 struct Edge{ int u,v,dis; }edge[maxm];
14 vector<int>to[maxn];
15 inline bool cmp(const Edge&a,const Edge&b){
16     return a.dis<b.dis;
17 }
18 inline int getf(int n){
19     if(fa[n]==n) return n;
20     fa[n]=getf(fa[n]);
21     return fa[n];
22 }
23 inline void Kruskal(){
24     for(int i=1;i<=(N<<1);i++) fa[i]=i;
25     sort(edge+1,edge+M+1,cmp);
26     int cnt=0;
27     for(int i=1;i<=M;i++){
28         int x=edge[i].u,y=edge[i].v,d=edge[i].dis;
29         int f1=getf(x),f2=getf(y);
30         if(f1!=f2){
31             n++;
32             D[n]=d;
33             fa[f1]=fa[f2]=n;
34             to[n].push_back(f1);
35             to[n].push_back(f2);
36             cnt++;
37             if(cnt==N-1) return;
38         }
39     }
40     return;
41 }
42 void Dfs(int x,int fa){
43     F[x][0]=fa;
44     Dep[x]=Dep[fa]+1;
45     for(int i=1;i<=maxl;i++)
46         F[x][i]=F[F[x][i-1]][i-1];
47     for(int i=0;i<(int)to[x].size();i++)
48         Dfs(to[x][i],x);
49     return;
50 }
51 inline int LCA(int x,int y){
52     if(Dep[x]<Dep[y]) swap(x,y);
53     for(int i=maxl;i>=0;i--){
54         if(Dep[F[x][i]]>=Dep[y])
55             x=F[x][i];
56     }
57     if(x==y) return x;
58     for(int i=maxl;i>=0;i--)
59         if(F[x][i]!=F[y][i])
60             x=F[x][i],y=F[y][i];
61     return F[x][0];
62 }
63 int main(){
64     N=rd(); M=rd(); K=rd();
65     n=N;
66     for(int i=1;i<=M;i++){
67         edge[i].u=rd();
68         edge[i].v=rd();
69         edge[i].dis=rd();
70     }
71     Kruskal();
72     Dfs(n,0);
73     while(K--){
74         A=rd(); B=rd();
75         printf("%d\n",D[LCA(A,B)]);
76     }
77     return 0;
78 }
View Code

 


By:AlenaNuna

 

上一篇:P1137只有60分


下一篇:【noip2015】