洛谷 P1967 货车运输

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

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

输出格式：

输出文件名为 truck.out。
输出共有 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
说明
对于 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

---

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
int n,m,q,fa[10001],dp[10005],f[10005][21],mn[10005][21];
struct xd
{
    int x,y,val;
}a[50001];
struct node
{
    int z,val;
}tmp;
vector<node>v[50005];
bool comp(xd c,xd d)
{
    return c.val>d.val;
}
int fnd(int x)
{
    if(fa[x]==x)
        return x;
    return fa[x]=fnd(fa[x]);
}
void dfs(int u,int step)
{
    dp[u]=step;
    for(int i=0;i<v[u].size();i++)
        if(!dp[v[u][i].z])
        {
            f[v[u][i].z][0]=u;
            mn[v[u][i].z][0]=v[u][i].val;
            dfs(v[u][i].z,step+1);
        }
}
int lca(int a,int b)
{
    int minn=1e9+7;
    if(dp[a]<dp[b])
        swap(a,b);
    for(int i=15;i>=0;i--)
        if(dp[a]>=dp[b]+(1<<i))
            minn=min(minn,mn[a][i]),a=f[a][i];
    if(a==b)
        return minn;
    for(int i=15;i>=0;i--)
        if(f[a][i]!=f[b][i])
            minn=min(minn,min(mn[a][i],mn[b][i])),a=f[a][i],b=f[b][i];
    return min(minn,min(mn[a][0],mn[b][0]));
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].val);
    sort(a+1,a+m+1,comp);
    for(int i=1;i<=n;i++)
        fa[i]=i;
    for(int i=1;i<=m;i++)
        if(fnd(a[i].x)!=fnd(a[i].y))
        {
            fa[fnd(a[i].y)]=fnd(a[i].x);
            tmp.val=a[i].val;
            tmp.z=a[i].y;
            v[a[i].x].push_back(tmp);
            tmp.z=a[i].x;
            v[a[i].y].push_back(tmp);
        }
    dfs(1,1);
    for(int i=0;i<=15;i++)
        for(int j=1;j<=n;j++)
        {
            f[j][i+1]=f[f[j][i]][i];
            mn[j][i+1]=min(mn[j][i],mn[f[j][i]][i]);
        }
    scanf("%d",&q);
    int x,y;
    while(q--)
    {
        scanf("%d%d",&x,&y);
        if(fnd(x)!=fnd(y))
            printf("-1\n");
        else
            printf("%d\n",lca(x,y));
    }
    return 0;
}