0pts求助
查看原帖
0pts求助
1314528
Oscar111111楼主2025/8/2 19:03
#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef pair<int,int> PII;
const int N=5e4+500;
const int inf=0x7fffffff;
int n,m,x,y,q,cnt,fa[N],dep[N],up[N][21],mn[N][21],ans,vis[N];
struct node{
	int u,v,w;
}a[N];
vector<PII>to[N];
int find(int x){
	if(fa[x]!=x) fa[x]=find(fa[x]);
	return fa[x];
}
bool cmp(node a,node b){
	return a.w>b.w;
}
void dfs(int x,int pre){
	vis[x]=1;
	dep[x]=dep[pre]+1;
	up[x][0]=pre;
	mn[x][0]=inf;
	for(int i=1;i<=20;i++) up[x][i]=up[up[x][i-1]][i-1],mn[x][i]=min(mn[x][i-1],mn[up[x][i-1]][i-1]);
	for(auto y:to[x]){
		if(y.first==pre) continue;
		mn[y.first][0]=y.second;
		dfs(y.first,x);
	}
}
int lca(int x,int y){
	if(find(x)!=find(y)) return -1;
	int ans=inf;
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=20;i>=0;i--){
		if(dep[x]-(1<<i)>=dep[y])
			x=up[x][i],ans=min(ans,mn[x][i]);
	}
	if(x==y) return x;
	for(int i=20;i>=0;i--){
		if(up[x][i]!=up[y][i])
			x=up[x][i],y=up[y][i],ans=min(ans,min(mn[x][i],mn[y][i]));
	}
	ans=min(ans,min(mn[x][0],mn[y][0]));
	return ans;
}
void kruskal(){
	int num=0;
	for(int i=1;i<=n;i++) fa[i]=i;
	sort(a+1,a+m+1,cmp);
	for(int i=1;i<=m;i++){
		if(find(a[i].u)!=find(a[i].v)){
			cnt++;
			num++;
			fa[find(a[i].u)]=find(a[i].v);
			to[a[i].u].push_back({a[i].v,a[i].w});
			to[a[i].v].push_back({a[i].u,a[i].w});
		}
		if(num==n-1) break;
	}
	for(int i=1;i<=cnt;i++){
    	if(!vis[i]){
    	    int f=find(i);
    	    dfs(f,0);
    	}
    }
	return;
}
signed main(){
	cin>>n>>m;
	for(int i=1;i<=m;i++) cin>>a[i].u>>a[i].v>>a[i].w;
	kruskal();
	cin>>q;
	while(q--){
		cin>>x>>y;
		cout<<lca(x,y)<<endl;
	}
	return 0;
}

好奇怪,其他警示的帖子都有了还是0

2025/8/2 19:03
加载中...