10分求助呜呜谢谢大家了真的不知道问题在哪里呜呜
查看原帖
10分求助呜呜谢谢大家了真的不知道问题在哪里呜呜
229446
ephemere楼主2020/9/8 18:51
#include<bits/stdc++.h>
using namespace std;

const int N=1e4+5,M=5e4+5;
int n,m,te,tail[N],f[N],dep[N],F[N][14],D[N][14];//log2(N)=13.2877
struct e_
{
	int u,v,w;
	friend bool operator<(e_ x,e_ y)
	{
		return x.w>y.w;
	}
}e[M];
struct E_
{
	int v,w,pre;
}E[M<<1];

int find(int x)
{
	return f[x]!=x?f[x]=find(f[x]):x;
}
inline void add(int u,int v,int w)
{
	E[++te]=(E_){v,w,tail[u]};
	tail[u]=te;
}
void dfs(int u)
{
	for(int i=tail[u];i;i=E[i].pre)
	{
		int v=E[i].v,w=E[i].w;
		if(v==F[u][0]) continue;
		
		F[v][0]=u;D[v][0]=w;
		dep[v]=dep[u]+1;
		for(int j=1;(1<<j)<=dep[v]&&j<=13;++j)
		F[v][j]=F[F[v][j-1]][j-1],D[v][j]=min(D[v][j-1],D[F[v][j-1]][j-1]);

		dfs(v);
	}
}
int LCA_(int u,int v)
{
	if(find(u)!=find(v)) return -1;
	
	int res=1e5+5;
	if(dep[u]<dep[v]) swap(u,v);
	
	for(int j=13;j>=0;--j)
	if(dep[v]+(1<<j)<=dep[u])
	{
		res=min(res,D[u][j]),u=F[u][j];
	}
	if(u==v) return res;
	
	for(int j=13;j>=0;--j)
	if(F[u][j]!=F[v][j]) res=min(res,min(D[u][j],D[v][j])),u=D[u][j],v=D[v][j];
	
	res=min(res,min(D[u][0],D[v][0]));
	return res;
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;++i) f[i]=i;
	for(int i=1;i<=m;++i) scanf("%d %d %d",&e[i].u,&e[i].v,&e[i].w);
	sort(e+1,e+m+1);
	for(int i=1;i<=m;++i)
	{
		int x=find(e[i].u),y=find(e[i].v);
		
		if(x==y) continue;
			
		f[x]=y;f[0]--;
		add(e[i].u,e[i].v,e[i].w);
		add(e[i].v,e[i].u,e[i].w);
	}
	
	for(int i=1;i<=n;++i)
	if(find(i)==i) dfs(i);
	
	scanf("%d",&m);
	for(int i=1,x,y;i<=m;++i) scanf("%d %d",&x,&y),printf("%d\n",LCA_(x,y));
}
2020/9/8 18:51
加载中...