最小生成树+LCA【洛谷 P2245】 星际导航

【洛谷 P2245】 星际导航

货车运输。最小瓶颈路。

code：

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

const int wx=1000017;

inline int read(){
int sum=0,f=1; char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0';ch=getchar();}
return sum*f;
}

int fa[wx],dep[wx];
int f[wx][23],dis[wx][23];
int head[wx];
int n,m,num,q;

struct node{
int l,r,d;
friend bool operator < (const node& a,const node& b){
return a.d<b.d;
}
}a[wx*2];

struct e{
int nxt,to,dis;
}edge[wx*2];

void add(int from,int to,int dis){
edge[++num].nxt=head[from];
edge[num].to=to;
edge[num].dis=dis;
head[from]=num;
}

int find(int x){
if(x==fa[x])return x;
return fa[x]=find(fa[x]);
}

void build(int flag){
add(a[flag].l,a[flag].r,a[flag].d);
add(a[flag].r,a[flag].l,a[flag].d);
}

void Kruskal(){
for(int i=1;i<=n;i++)fa[i]=i;
sort(a+1,a+1+m);
for(int i=1;i<=m;i++){
if(find(a[i].l)!=find(a[i].r)){
fa[find(a[i].l)]=find(a[i].r);
build(i);
}
}
}

void dfs(int u,int father){
dep[u]=dep[father]+1;
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].to;
if(v==father)continue;
f[v][0]=u;
dis[v][0]=edge[i].dis;
dfs(v,u);
}
}

void pre(){
for(int j=1;j<=21;j++){
for(int i=1;i<=n;i++){
f[i][j]=f[f[i][j-1]][j-1];
dis[i][j]=max(dis[i][j-1],dis[f[i][j-1]][j-1]);
}
}
}

int FFF(int x,int y){
int re=0;
if(dep[x]<dep[y])swap(x,y);
for(int i=21;i>=0;i--){
if(dep[f[x][i]]>=dep[y]){
re=max(re,dis[x][i]);
x=f[x][i];
}
}
if(x==y)return re;
for(int i=21;i>=0;i--){
if(f[x][i]!=f[y][i]){
re=max(re,dis[x][i]);
re=max(re,dis[y][i]);
x=f[x][i]; y=f[y][i];
}
}
return max(re,max(dis[x][0],dis[y][0]));
}

int main(){
n=read(); m=read();
for(int i=1;i<=m;i++){
int x,y,z;
x=read(); y=read(); z=read();
a[i].l=x; a[i].r=y; a[i].d=z;
}
Kruskal();
dfs(1,0); pre();
q=read();
for(int i=1;i<=q;i++){
int x,y;
x=read(); y=read();
FFF(x,y);
if(find(x)==find(y)) printf("%d\n",FFF(x,y));
else printf("impossible\n");
}
return 0;
}