10*WA求助
查看原帖
10*WA求助
262209
Magic_Kaito楼主2020/10/17 19:52

如题

#include <bits/stdc++.h>
#define int long long
#define N 1000005
using namespace std;
struct edge
{
	int x,y,w;
}E[N];
int _id,p[N],nt[N],w[N],b[N],n,m,fa[N],f[N][25],dist[N][25],deep[N],flag[N];
void add(int x,int y,int z)
{
	++_id;
	b[_id]=y;w[_id]=z;nt[_id]=p[x];p[x]=_id;
}
int _find(int x)
{
	if(fa[x]==x)return x;
	return fa[x]=_find(fa[x]);
}
bool cmp(edge s,edge t)
{
	return s.w>t.w;
}
void dfs(int x)
{
	flag[x]=1;
	for(int i=1;i<=20;i++)
	{
		if(deep[x]<(1<<i))break;
		f[x][i]=f[f[x][i-1]][i-1];
		dist[x][i]=min(dist[x][i-1],dist[f[x][i-1]][i-1]);
	}
	int e=p[x];
	while(e>0)
	{
		int k=b[e];
		if(!flag[k])
		{
			f[k][0]=x;
			dist[k][0]=w[e];
			deep[k]=deep[x]+1;
			dfs(k);
		}
		e=nt[e];
	}
}
int lca(int x,int y)
{
	if(deep[x]<deep[y])swap(x,y);
	int tt=deep[x]-deep[y];
	for(int i=0;i<=20;i++)
	{
		if(tt&(1<<i))
		{
			x=f[x][i];
		}
	}
	for(int i=20;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];y=f[y][i];
		}
	}
	if(x==y)return x;
	else return f[x][0];
}
int ct(int x,int ff)
{
	int tt=deep[x]-deep[ff];
	int minv=1<<30;
	for(int i=1;i<=20;i++)
	{
		if(tt&(1<<i))
		{
			minv=min(minv,dist[x][i]);
			x=f[x][i];
		}
	}
	return minv;
}
signed main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++)cin>>E[i].x>>E[i].y>>E[i].w;
    sort(E+1,E+1+m,cmp);
    int num=0;
    for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		int x=E[i].x;
		int y=E[i].y;
		int fx=_find(x);
		int fy=_find(y);
		if(fx!=fy)
		{
			++num;
			add(x,y,E[i].w);
			add(y,x,E[i].w);
			fa[fx]=fy;
			if(num==n-1)break;
		}
	} 
	for(int i=1;i<=n;i++)if(flag[i]==0)dfs(i);
	int q;
	cin>>q;
	for(int i=1;i<=q;i++)
	{
		int x,y;
		cin>>x>>y;
		if(_find(x)!=_find(y))cout<<-1<<endl;
		else
		{
			int father=lca(x,y);
			int aa=ct(x,father);
			int bb=ct(y,father);
			cout<<min(aa,bb)<<endl;
		}
	}
	return 0;
}
2020/10/17 19:52
加载中...