货车运输

货车运输

题目描述： 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。
样例输入： 4 3 1 2 4 2 3 3 3 1 1 3 1 3 1 4 1 3
样例输出： 3
-1 3
数据范围及提示： 对于 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。

来源：2013年NOIP全国联赛提高组
算法使用：最大生成树+LCA（倍增）

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=1000000;
struct node
{
    int x;
    int y;
    int v;
    int next;
}e[maxn],ed[maxn*3];
int n,m,q,x,y,sum,ans;
int father[maxn],fa[maxn][25],dis[maxn][25],deep[maxn],first[maxn];
int find(int x)
{
    if(x!=father[x])
    father[x]=find(father[x]);
    return father[x];
}
void unionn(int x,int y)
{
    father[y]=x;
}
int cmp(node x,node y)
{
    return x.v>y.v;
}
void add_edge(int x,int y,int v)
{
    sum++;
    ed[sum].x=x;
    ed[sum].y=y;
    ed[sum].v=v;
    ed[sum].next=first[x];
    first[x]=sum;
}
void fkruskal()
{
    int tot=0;
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        int f1=find(e[i].x);
        int f2=find(e[i].y);
        if(f1!=f2)
        {
            unionn(f1,f2);
            add_edge(e[i].x,e[i].y,e[i].v);
            add_edge(e[i].y,e[i].x,e[i].v);
            tot++;
        }
        if(tot==m-1) break;
    }
}
void build(int u)
{
    for(int i=first[u];i;i=ed[i].next)
    {
        int v=ed[i].y;
        if(!fa[v][0])
        {
            fa[v][0]=u;
            deep[v]=deep[u]+1;
            dis[v][0]=ed[i].v;
            build(v);
        }
    } 
}
void init()
{
    for(int j=1;j<=18;j++)
      for(int i=1;i<=n;i++)
      {
        fa[i][j]=fa[fa[i][j-1]][j-1];
        dis[i][j]=min(dis[i][j-1],dis[fa[i][j-1]][j-1]);
      }
}
int LCA(int x,int y)
{
    ans=100000000;
    if(deep[x]<deep[y])
    swap(x,y);
    if(deep[x]!=deep[y])
    {
        int d=deep[x]-deep[y];
        for(int i=0;d;i++)
        {
            if(d&1)
            {
                ans=min(ans,dis[x][i]);
                x=fa[x][i]; 
            }
            d>>=1;
        }
    }
    if(x==y)
    return ans;
    for(int j=20;j>=0;j--)
    {
        if(fa[x][j]!=fa[y][j])
        {
            ans=min(ans,dis[x][j]);
            ans=min(ans,dis[y][j]);
            x=fa[x][j];
            y=fa[y][j];
        }
    }
    ans=min(ans,min(dis[x][0],dis[y][0]));
    return ans;
}
int main()
{
    memset(dis,9,sizeof(dis));
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    cin>>e[i].x>>e[i].y>>e[i].v;
    for(int i=1;i<=n;i++)
    father[i]=i;
    fkruskal();
    fa[1][0]=1;
    dis[1][0]=100000000;
    deep[1]=0;
    build(1);
    init();
    cin>>q;
    for(int i=1;i<=q;i++)
    {
        cin>>x>>y;
        if(find(x)!=find(y))
        cout<<-1<<endl;
        else
        cout<<LCA(x,y)<<endl;
    }
    return 0;
}