[算法模板]Kruskal重构树
kruskal重构树是一个很常用的图论算法。主要用于解决u->v所有路径上最长边的最小值,就是找到\(u->v\)的一条路径,使路径上的最长边最小。
图片来自Kruskal重构树学习笔记+BZOJ3732 Network
从上图我们可以看出,kruskal重构树有以下特质:
- 每个原图上的节点一一对应重构树上的叶子节点。
- 重构树上每一个其他节点(正方形)代表原图上的一个边,有点权。
- 重构树是一棵二叉树。
- 重构树是一个二叉堆。(所以两个叶子节点的LCA即为路径上的最大边)
那如何建树呢?显然,在kruskal基础上搞一搞就行了:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
#define maxn 25000
struct gg{
int u,v,w;
}side1[maxn*2];
vector<int> side2[maxn*4];
bool cop(gg x,gg y){return x.w<y.w;}
int ncnt,num[maxn*4],n,m,k,head[maxn],cnt,dep[maxn*4],f[maxn*4][21],fa[maxn*4];
int get(int x){
if(fa[x]==x)return x;
fa[x]=get(fa[x]);
return fa[x];
}
void uni(int x,int y,int w){
int gx=get(x),gy=get(y);
if(gx==gy)return;
ncnt++;num[ncnt]=w;
side2[ncnt].push_back(gx);side2[ncnt].push_back(gy);side2[gx].push_back(ncnt);side2[gy].push_back(ncnt);
fa[gx]=fa[gy]=fa[ncnt]=ncnt;
return;
}
void dfs(int u,int g){
dep[u]=dep[g]+1;f[u][0]=g;
for(int i=1;i<=20;i++)f[u][i]=f[f[u][i-1]][i-1];
for(int i=0;i<(int)side2[u].size();i++){
int v=side2[u][i];if(v==g)continue;
dfs(v,u);
}
return;
}
int lca(int u,int v){
if(dep[u]<dep[v])swap(u,v);
for(int i=20;i>=0;i--)if(dep[f[u][i]]>=dep[v])u=f[u][i];
if(u==v)return u;
for(int i=20;i>=0;i--)if(f[u][i]!=f[v][i]){u=f[u][i];v=f[v][i];}
return f[u][0];
}
int main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=m;i++){
int u,v,w;scanf("%d%d%d",&u,&v,&w);
side1[i]=(gg){u,v,w};
}
for(int i=0;i<=n;i++){fa[i]=i;}
ncnt=n;
sort(side1+1,side1+1+m,cop);
for(int i=1;i<=m;i++){
if(get(side1[i].u)==get(side1[i].v))continue;
uni(get(side1[i].u),get(side1[i].v),side1[i].w);
}
dfs(ncnt,0);
for(int i=1;i<=k;i++){
int a,b;scanf("%d%d",&a,&b);
printf("%d\n",num[lca(a,b)]);
}
return 0;
}