CodeVS3287[NOIP2013] 货车运输【Kruskal+倍增求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），另一方面，求出了这棵树过后，我们还要算出树上两点间的路径上边权的最小值。因为树上亮点的边权总是u->LCA(U,V)->v，我们就可以利用倍增求LCA将这条路上边权的最小值求出来。为此目的我们需要维护：
len[u][p]：节点u向上走2^p步的路上的最小边权
anc[u][p]：节点u向上走2^p步走到的节点的编号
具体地操作是这样（求LCA+求最小边权）：
1.让u往上跳，使得u、v位于同一深度，此时统计min(len[u][p])；
2.让u、v一起往上跳，直到两者跳到同一点，此时统计min(len[u][p],len[v][p])；
把上面统计的两个值的最小值输出就行了。
还有就是什么情况输出-1，显然如果u、v不连通的话就直接输出-1。我们前面进行Kruskal中的并查集的father数组就存储了这一条信息。
代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=10000,M=50000,P=20;
const int inf=1e5+5;
struct edge
{
    int u,v,w;
    int next;
}p[2*M+5],ed[2*M+5];
int head[N+5],num;
int father[N+5];
int anc[N+5][P+1],len[N+5][P+1],dep[N+5];
int n,m,q;
void build(int u,int v,int w)
{
    ed[++num].u=u;
    ed[num].v=v;
    ed[num].w=w;
    ed[num].next=head[u];
    head[u]=num;
}
bool cmp(edge a,edge b)
{
    return a.w>b.w;
}
int find(int a)
{
    return father[a]==a?a:father[a]=find(father[a]);
}
void kruskal()//构建最大生成树 
{
    sort(p+1,p+1+2*m,cmp);
    int tot=0;
    for(int i=1;i<=2*m;i++)
    {
        int u=find(p[i].u),v=find(p[i].v);
        if(u==v)continue;
        father[v]=u;
        build(p[i].u,p[i].v,p[i].w);
        build(p[i].v,p[i].u,p[i].w);
        if(++tot==n-1)return ;
    }
}
void dfs(int u,int f)
{
    dep[u]=dep[f]+1;
    anc[u][0]=f;
    for(int p=1;p<=P;p++)
    len[u][p]=min(len[u][p-1],len[anc[u][p-1]][p-1]),
    anc[u][p]=anc[anc[u][p-1]][p-1];
    for(int i=head[u];i!=-1;i=ed[i].next)
    {
        int v=ed[i].v;
        if(v==f)continue;
        len[v][0]=ed[i].w;
        dfs(v,u);
    }
}
int vmin(int u,int v)
{
    int ret=inf;
    if(dep[v]>dep[u])swap(u,v);
    int t=dep[u]-dep[v];
    for(int p=0;p<=P;p++)//u跳到和v一样高 
    if(t&(1<<p))
    {
        ret=min(ret,len[u][p]);
        u=anc[u][p];
    }
    if(u==v)return ret;
    for(int p=P;p>=0;p--)//u、v一起往上跳 
    if(anc[u][p]!=anc[v][p])
    {
        ret=min(ret,len[u][p]);
        ret=min(ret,len[v][p]);
        u=anc[u][p],v=anc[v][p];
    }
    ret=min(ret,len[u][0]);
    ret=min(ret,len[v][0]);
    return ret;
}
int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)father[i]=i;
    for(int i=1;i<=n;i++)
    for(int p=1;p<=P;p++)len[n][p]=inf;
    for(int i=1;i<=m;i++)scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].w),p[i+m]=p[i];
    kruskal();
    for(int i=1;i<=n;i++)if(dep[i]==0)dfs(i,i);
    scanf("%d",&q);
    while(q--)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        if(find(u)!=find(v))printf("-1\n");
        else printf("%d\n",vmin(u,v));
    }
    return 0;
}