悬赏0.114514津巴布韦币倍增求调
查看原帖
悬赏0.114514津巴布韦币倍增求调
267428
Access57楼主2022/10/6 19:53
#include<bits/stdc++.h>
using namespace std;
int n,g,q;
int f[50010][23];
int m[50010][23];
int depth[50010];
bool vis[10010];
vector<pair<int,int> > z[10010];	
//DAG
struct edge
{
	int f,t,v;
}e[50010];
//union find
int fa[10010];
void init_union_find()
{
	for(int i=1;i<=n;i++) fa[i]=i;
}
int find(int x)
{
	if(fa[x]!=x) return fa[x]=find(fa[x]);
	else return x;
}
void merge(int a,int b)
{
	a=find(a),b=find(b); 
	if(a==b) return;
	if(rand()%2) swap(a,b);
	fa[a]=b;
}
//LCA & MAXN

//INIT
void dfs(int now,int fa)
{
	vis[now]=1;
	depth[now]=depth[fa]+1;
	for(int i=0;i<z[now].size();i++) if(z[now][i].first!=fa)
	{
		int to=z[now][i].first;
		int val=z[now][i].second;
		
		f[to][0]=now,m[to][0]=val;
		
		for(int j=1;j<23;j++)
		{
			f[to][j]=f[f[to][j-1]][j-1],
			m[to][j]=min(m[to][j-1],m[f[to][j-1]][j-1]);
		}
		dfs(to,now);
	}
}
//find LCA & MAXN
int getmax(int n1,int n2)
{
	int ans=1e9;
	if(find(n1)!=find(n2)) return -1;
	if(depth[n1]<depth[n2]) swap(n1,n2);
	
	for(int i=22;i>=0;i--) 
		if(depth[f[n1][i]]>=depth[n2]) 
			ans=min(ans,m[n1][i]),n1=f[n1][i];
	if(n1==n2) 
	{
		return ans;
	}

	for(int i=22;i>=0;i--) if(f[n1][i]!=f[n2][i])
	{
		n1=f[n1][i],n2=f[n2][i];
		
		ans=min(ans,min(m[n1][i],m[n2][i]));
	}
	ans=min(ans,m[n1][0]);
	return ans;
}
//kruskal
bool compare(const edge &n1,const edge &n2)
{
	return n1.v>n2.v;
}
int main()
{
	cin>>n>>g;
	init_union_find();
	for(int i=0;i<g;i++)
		cin>>e[i].f>>e[i].t>>e[i].v;
	sort(e,e+g,compare);
	for(int i=0;i<g;i++)
	{
		if(find(e[i].f)==find(e[i].t)) continue;
		merge(e[i].f,e[i].t);
		z[e[i].f].push_back(make_pair(e[i].t,e[i].v));
		z[e[i].t].push_back(make_pair(e[i].f,e[i].v));
	}
	depth[1]=1;
	for(int i=1;i<=n;i++) if(!vis[i]) dfs(i,0);
	cin>>q;
	while(q--)
	{
		int a,b;
		cin>>a>>b;
		cout<<getmax(a,b)<<"\n";
	}
}
2022/10/6 19:53
加载中...