codevs 3287 货车运输(树上倍增,最大生成树)

题目描述 Description

A 国有 n 座城市，编号从 1 到 n，城市之间有 m 条双向道路。每一条道路对车辆都有重量限制，简称限重。现在有 q 辆货车在运输货物，司机们想知道每辆车在不超过车辆限重的情况下，最多能运多重的货物。

输入描述 Input Description

第一行有两个用一个空格隔开的整数 n，m，表示 A 国有 n 座城市和 m 条道路。
接下来 m 行每行 3 个整数 x、y、z，每两个整数之间用一个空格隔开，表示从 x 号城市到 y 号城市有一条限重为 z 的道路。注意：x 不等于 y，两座城市之间可能有多条道路。
接下来一行有一个整数 q，表示有 q 辆货车需要运货。
接下来 q 行，每行两个整数 x、y，之间用一个空格隔开，表示一辆货车需要从 x 城市运输货物到 y 城市，注意：x 不等于 y。

输出描述 Output Description

输出共有 q 行，每行一个整数，表示对于每一辆货车，它的最大载重是多少。如果货车不能到达目的地，输出-1。

样例输入 Sample Input

4 3 
1 2 4 
2 3 3 
3 1 1 
3
1 3 
1 4 
1 3

样例输出 Sample Output

3
-1
3

数据范围及提示 Data Size & Hint

对于 30%的数据，0 < n < 1,000，0 < m < 10,000，0 < q < 1,000； 
对于 60%的数据，0 < n < 1,000，0 < m < 50,000，0 < q < 1,000； 
对于 100%的数据，0 < n < 10,000，0 < m < 50,000，0 < q < 30,000，0 ≤ z ≤ 100,000。

题解：

先求最大生成树，然后每对点，在最大生成树上找出lca，并用与lca类似的倍增法预处理d[x][i](结点x向上走2^i个结点的路径中小的一段)。

注意点可能不连通，所以在求最大生成树时无法判断要取多少条边，不过直接对每条边判断端点是否连通(并查集)就行了，不连通就加进去。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
typedef vector<int> VI;
typedef long long ll;
typedef pair<int,int> PII;
const ll mod=1000000007;
const int MAXN=10000+100;
const int MAXM=50000+100;
const int INF=0x3ffffff;
int head[MAXN],pa[MAXN],fa[MAXN][16],deep[MAXN],d[MAXN][16];
bool vis[MAXN];
struct fun
{
	int u,v,w;
	bool operator <(const fun& b)const{
		return w>b.w;
	}
}a[MAXM];
struct node
{
	int to,next,w;
}edge[MAXM*2];
int find(int x)
{
	return pa[x]==x? x:pa[x]=find(pa[x]);
}
int tol=0;
void add(int u,int v,int w)
{
	edge[++tol].to=v,edge[tol].next=head[u],head[u]=tol,edge[tol].w=w;
	edge[++tol].to=u,edge[tol].next=head[v],head[v]=tol,edge[tol].w=w;
}
void dfs(int x)
{
	vis[x]=1;
	for(int i=1;i<15;i++)
	{
		if(deep[x]<(1<<i)) break;
		fa[x][i]=fa[fa[x][i-1]][i-1];
		d[x][i]=min(d[x][i-1],d[fa[x][i-1]][i-1]);  //类似RMQ进行更新 
	}
	for(int i=head[x];i;i=edge[i].next)
	{
		if(vis[edge[i].to]) continue;
		fa[edge[i].to][0]=x;
		deep[edge[i].to]=deep[x]+1;
		d[edge[i].to][0]=edge[i].w;
		dfs(edge[i].to);
	}
}
int lca(int x,int y)
{
	if(deep[x]<deep[y]) swap(x,y);
	int d=deep[x]-deep[y];
	rep(i,0,15)
		if(d&(1<<i)) x=fa[x][i];
	if(x==y) return x;
	per(i,0,15)
	if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int solve(int x,int y)
{
	int t=lca(x,y);
	int ans1=INF,ans2=INF;
	int di=deep[x]-deep[t];
	rep(i,0,15)
	if(di&(1<<i)) ans1=min(ans1,d[x][i]),x=fa[x][i];  //类似倍增法往上跳 
	di=deep[y]-deep[t];
	rep(i,0,15)
	if(di&(1<<i)) ans2=min(ans2,d[y][i]),y=fa[y][i];
	return min(ans1,ans2);
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	rep(i,1,n+1) pa[i]=i;
	rep(i,1,m+1)
	{
		int u,v,w;
		scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
	}
	sort(a+1,a+m+1);
	rep(i,1,m+1)          //求最大生成树,然后再建图 
	{
		int fu=find(a[i].u),fv=find(a[i].v);
		if(fu!=fv)
		{
			pa[fu]=fv;
			add(a[i].u,a[i].v,a[i].w);
		}
	}
	rep(i,1,n+1) if(!vis[i]) dfs(i);
	int q;
	scanf("%d",&q);
	while(q--)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		if(find(u)!=find(v)) puts("-1");
		else printf("%d\n",solve(u,v));
	}
	return 0;
}