codevs 3287货车运输 （最大生成树+lca）

题目描述 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。

思路：最优解一定是在原图的最大生成树上，先使用kruskal建一棵最大生成树并保存边的信息，然后对树dfs得到节点的深度以及dis[]，然后lca的时候每次都取小。

代码如下

#include <cstdio>  
#include <cstdlib>  
#include <cstring>  
#include <cmath>  
#include <algorithm>  
#include <iostream>  
#include <string>  
#include <vector>  
using namespace std;  

const int N=10100;
const int M=50005;
int head[N];
int dis[N],deep[N],fa[N],pre[N];
int n,m,q,tol;
bool vis[N];
struct Edge1
{
    int from,to,cost;
}edge1[M];

struct Edge2//链式向前星存边
{
    int to,cost;
    int next;
}edge2[M];

bool cmp(Edge1 a,Edge1 b)
{
    return a.cost>b.cost;
}

void init()
{
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++) 
    pre[i]=i;

    tol=0;
}

void addedge(int u,int v,int cost)
{
    edge2[tol].to=v;
    edge2[tol].cost=cost;
    edge2[tol].next=head[u];
    head[u]=tol++;
}

int find(int x)
{
   int r=x;
   while (pre[r]!=r)
   r=pre[r];
   int i=x; int j;
   while(i!=r)
   {
       j=pre[i];
       pre[i]=r;
       i=j;
   }
   return r;
}
void dfs(int s)
{
    vis[s]=1;
    for(int i=head[s];i!=-1;i=edge2[i].next)
    {
        int to=edge2[i].to;
        if(!vis[to])
        {
            fa[to]=s;
            deep[to]=deep[s]+1;
            dis[to]=edge2[i].cost;
            dfs(to);
        }
    }
}
int lca(int x,int y)
{
    int minn=1e5+1;//利用lca找路径上的最小值 
    if(deep[x]>deep[y]) swap(x,y);
    while(deep[x]!=deep[y])
    {
        if(dis[y]<minn) minn=dis[y];
        y=fa[y];
    }
    while(x!=y)
    {
        if(dis[x]<minn) minn=dis[x];
        if(dis[y]<minn) minn=dis[y];
        x=fa[x];
        y=fa[y];
    }
    return minn;
}
int main()
{
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=m;i++)
    scanf("%d%d%d",&edge1[i].from,&edge1[i].to,&edge1[i].cost);
    sort(edge1+1,edge1+m+1,cmp);//用Kruskal生成最大生成树 
    for(int i=1;i<=m;i++)
    {
        int u=find(edge1[i].from);
        int v=find(edge1[i].to);
        if(u!=v)
        {
            pre[u]=v;
            addedge(edge1[i].from,edge1[i].to,edge1[i].cost);
            addedge(edge1[i].to,edge1[i].from,edge1[i].cost);
        }
    }

    dfs(1);//遍历树，得到节点的深度和dis
    scanf("%d",&q);
    for(int i=1;i<=q;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        if(find(u)!=find(v))//用并查集来判断是否有联通性
        printf("-1\n");
        else  
        printf("%d\n",lca(u,v));
    }
    return 0;
}