洛谷-1967 货车运输

题目描述
A国有n座城市，编号从 1到n，城市之间有 m 条双向道路。每一条道路对车辆都有重量限制，简称限重。现在有 q 辆货车在运输货物， 司机们想知道每辆车在不超过车辆限重的情况下，最多能运多重的货物。
输入输出格式
输入格式：
第一行有两个用一个空格隔开的整数n,m，表示 A 国有n 座城市和 m 条道路。
接下来 m行每行3个整数 x,y,z，每两个整数之间用一个空格隔开，表示从 x号城市到y 号城市有一条限重为 z 的道路。注意： x 不等于 y，两座城市之间可能有多条道路 。
接下来一行有一个整数 q，表示有 q 辆货车需要运货。
接下来 q 行，每行两个整数 x、y，之间用一个空格隔开，表示一辆货车需要从 x 城市运输货物到 y 城市，注意： x 不等于 y
输出格式：
共有 q 行，每行一个整数，表示对于每一辆货车，它的最大载重是多少。如果货车不能到达目的地，输出−1。

输入输出样例
输入样例#1：
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3

输出样例#1：
3
-1
3

说明
对于 100%的数据，0<n<10,000,0<m<50,000,0<q<30,000,0≤z≤100,000

解释：仔细分析一下，就发现满足题意的任意一点路径形成的图就是一个最大生成树（不知道怎么证明），然后我们在最大生成树上求公共祖先路上的最小边权，多加一个倍增法的数组就可以维护好了，同时注意可能是森林。

#include<iostream>
#include<cstdio>
#include<algorithm>
#define INF 100000009
using namespace std;
struct yyy{
    int t,nex;
    int c;
}e[2*50001];
struct edge{
    int a,b;
    int c;
}E[2*50001];
int depth[20003],fa[20003][22],lg[20003],head[20003];
int H[20003]={0};
int father[20000]={0};
int faval[20000][22]={0};
int tot;
void init(){
    for(int i=1;i<=20000;i++) father[i]=i;
}
int find(int x){
    if(father[x]==x) return x;
    return father[x]=find(father[x]);
}
void merge(int x,int y){
    x=find(x);y=find(y);
    father[x]=y;
}
bool same(int x,int y){
    return find(x)==find(y);
}
void add(int x,int y,int c){
    e[++tot].t=y;
    e[tot].c=c;
    e[tot].nex=head[x];
    head[x]=tot;
}
bool cmp(edge &a,edge &b){
    return a.c>b.c;
}
void dfs(int f,int fath){
    depth[f]=depth[fath]+1;
    fa[f][0]=fath;
    for(int i=1;(1<<i)<=depth[f];i++){
        fa[f][i]=fa[fa[f][i-1]][i-1];
        faval[f][i]=min(faval[fa[f][i-1]][i-1],faval[f][i-1]);
    }
    for(int i=head[f];i;i=e[i].nex)
        if(e[i].t!=fath){
            faval[e[i].t][0]=e[i].c;
            dfs(e[i].t,f);
        }
}
int lca(int x,int y){
    int ret=INF;
    if(depth[x]<depth[y])
      swap(x,y);
    while(depth[x]>depth[y]){
        ret=min(ret,faval[x][lg[depth[x]-depth[y]]-1]);
        x=fa[x][lg[depth[x]-depth[y]]-1];
    }
    if(x==y) return ret;
    for(int k=lg[depth[x]]-1;k>=0;k--)
        if(fa[x][k]!=fa[y][k]){
            ret=min(ret,faval[x][k]);
            ret=min(ret,faval[y][k]);
            x=fa[x][k],y=fa[y][k];
        }
    ret=min(ret,faval[y][0]);
    ret=min(ret,faval[x][0]);
    return ret;
}
int n,m;
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&E[i].a,&E[i].b,&E[i].c);
    }
    sort(E+1,E+1+m,cmp);
    init();
    for(int i=1;i<=m;i++){
        if(!same(E[i].a,E[i].b)){
            merge(E[i].a,E[i].b);
            add(E[i].a,E[i].b,E[i].c);
            add(E[i].b,E[i].a,E[i].c);
        }
    }
    for(int i=1;i<=n;i++){
        int root=find(i);
        if(!H[root]){
            faval[root][0]=INF;
            dfs(root,0);
        }
        H[root]=1;
    }
    for(int i=1;i<=n;i++)
      lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    int T=0;scanf("%d",&T);
    for(int i=1;i<=T;i++){
        int x,y;  scanf("%d%d",&x,&y);
        if(!same(x,y)) printf("-1\n");
        else printf("%d\n",lca(x,y));
    }
    return 0;
}