计蒜客题目 货车运输

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+7;
struct edge{
	int u,v,w,next;
}e[maxn];
edge e1[maxn];
int cnt=0;
int n,m,q;
int head[maxn];
int pre[maxn][23];
int d[maxn];
int dist[maxn][23];
bool cmp(edge a,edge b){
	return a.w>b.w;
}
int father[maxn];
void dfs(int node){
	for(int i=head[node];i!=-1;i=e1[i].next){
		int v=e1[i].v;
		if(d[v]==-1){
		d[v]=d[node]+1;
		pre[v][0]=node;
		dist[v][0]=e1[i].w;
		dfs(v);
		}
	}
}
int LCA(int x,int y){
	int ans=0x3f3f3f3f;
	if(d[x]<d[y]) swap(x,y);
	for(int i=log2(d[x]-d[y]);i>=0;i--){
		if((1<<i)<=d[x]-d[y]) ans=min(ans,dist[x][i]),x=pre[x][i];
	}
	if(x==y) return ans;
	for(int i=log2(d[x]);i>=0;i--){
		if(pre[x][i]!=pre[y][i]){
			ans=min(ans,dist[x][i]);
			ans=min(ans,dist[y][i]);
			x=pre[x][i];
			y=pre[y][i];

		}
	}
	ans=min(ans,dist[x][0]);
	ans=min(ans,dist[y][0]);
	return ans;
}
int get(int a){
	if(father[a]==a) return a;
	else return father[a]=get(father[a]);
}
void merge(int a,int b){
	a=get(a);
	b=get(b);
	if(a!=b) father[a]=b;
}
void add(int u,int v,int w){
	e1[cnt].v=v;
	e1[cnt].u=u;
	e1[cnt].w=w;
	e1[cnt].next=head[u];
	head[u]=cnt++;
}
int main(){
	memset(head,-1,sizeof head);
	memset(d,-1,sizeof d);
	cin>>n>>m;
	for(int i=1;i<=n;i++) father[i]=i;
	for(int i=1;i<=m;i++){
		cin>>e[i].u>>e[i].v>>e[i].w;
	}
	sort(e+1,e+m+1,cmp);
	//for(int i=1;i<=n-1;i++) cout<<e[i].w<<endl;
	for(int i=1;i<=m;i++){
		int u=e[i].u;
		int v=e[i].v;
		int w=e[i].w;
		if(get(u)!=get(v)){
			merge(u,v);
			add(u,v,w);
			add(v,u,w);
		}
	}
	for(int i=1;i<=n;i++){
		if(d[i]==-1) dfs(i);
	}
		for(int i=1;i<=22;i++){
		for(int j=1;j<=n;j++){
			pre[j][i]=pre[pre[j][i-1]][i-1];
			dist[j][i]=min(dist[j][i-1],dist[pre[j][i-1]][i-1]);
		}
	}
	cin>>q;
	for(int i=1;i<=q;i++){
		int a,b;
		cin>>a>>b;
		if(get(a)!=get(b)) cout<<-1<<endl;
		else cout<<LCA(a,b)<<endl;
	}
}