「NOIP2013」货车运输--kruskal+树上ST

传送门

分析

∵两座城市之间可能有多条道路，∴我们只需要保留一条最大的边即可，用kruskal实现。
∵求最大载重是多少，∴用一个树上ST。
如果货车不能到达目的地，即起点和终点不在同一棵树上，并查集判断即可。

代码

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define full(a,b) memset(a,b,sizeof a)
#define N (10000+5)
#define E (50000+5)
int n,m,q;
int head[N],ecnt,to[2*N],nxt[2*N],cost[2*N];
int fa[N];
int f[N][25],dep[N],dist[N],vis[N];
int st[N][25];
struct node
{
	int u,v,cost;
} Edge[E];
void init()
{
	full(fa,-1);
	full(head,-1);
	full(st,INF);
}
void add(int u,int v,int w)
{
	to[++ecnt]=v;
	nxt[ecnt]=head[u];
	head[u]=ecnt;
	cost[ecnt]=w;
}
int fi(int x)
{
	if(fa[x]==-1) return x;
	return fa[x]=fi(fa[x]);
}
int cmp(node x,node y)
{
	return x.cost>y.cost;//选择边权大的 
}
void pre(int u,int fa)
{
	dep[u]=dep[fa]+1;
	vis[u]=1;
	f[u][0]=fa;
	for(int i=0; i<=20; i++)
	{
		f[u][i+1]=f[f[u][i]][i];
		st[u][i+1]=min(st[u][i],st[f[u][i]][i]);//随便初始化ST 
	}
	for(int i=head[u]; i!=-1; i=nxt[i])
	{
		int v=to[i];
		if(vis[v]) continue;
		dist[v]=dist[u]+cost[i];
		st[v][0]=cost[i];//初始化ST 
		pre(v,u);
	}
}
int LCA(int x,int y)
{
	int ans=INF;
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=20; i>=0; i--)
	{
		if(dep[f[x][i]]>=dep[y])
		{
			ans=min(ans,st[x][i]);//一边找LCA，一边统计min 
			x=f[x][i];
		}
		if(x==y) return ans;
	}
	for(int i=20; i>=0; i--)
		if(f[x][i]!=f[y][i])
		{
			ans=min(ans,min(st[x][i],st[y][i]));
			x=f[x][i],y=f[y][i];
		}
	ans=min(ans,min(st[x][0],st[y][0]));
	return ans;
}
void kruskal()//去重边 
{
	sort(Edge+1,Edge+1+m,cmp);
	int k=0;
	for(int i=1; i<=m; i++)
	{
		int x=fi(Edge[i].u),y=fi(Edge[i].v);
		if(x!=y)
		{
			fa[x]=y;
			add(Edge[i].u,Edge[i].v,Edge[i].cost);
			add(Edge[i].v,Edge[i].u,Edge[i].cost);
			if(++k==n-1) break;
		}
	}
}
int main()
{
//	freopen("1359: 「NOIP2013」货车运输.in","r",stdin);
//	freopen("1359: 「NOIP2013」货车运输.out","w",stdout);
	init();
	scanf("%d%d",&n,&m);
	for(int i=1; i<=m; i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		Edge[i]=node{x,y,z};
	}
	kruskal();
	for(int i=1; i<=n; i++)
		if(!vis[i]) pre(i,0);//可能有多棵树 
	scanf("%d",&q);
	for(int i=1; i<=q; i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		if(fi(x)!=fi(y)) printf("-1\n");//不在一棵树上 
		else printf("%d\n",LCA(x,y));//树上ST结果 
	}
	return 0;
}