洛谷 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。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define MAXN 10050
#define MAXM 50050
int head[MAXN],fa[MAXN][20],dep[MAXN],tot,sum[MAXN][20],dad[MAXN];
struct Edge {
    int from,to,value;
} ee[MAXM];
struct EDGE {
    int next,to,value;
} e[MAXM<<1];
int n,m;
inline int read(int &x) {
    x=0;
    int f=1;
    char c=getchar();
    while(c>'9'||c<'0') {
        if(c=='-') f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9') {
        x=x*10+c-'0';
        c=getchar();
    }
    x*=f;
}

inline bool cmp(Edge a,Edge b) {
    return a.value > b.value;
}

int Find(int x) {
    if(x==dad[x]) return x;
    else return dad[x]=Find(dad[x]);
}

inline void Add_Edge(int u,int v,int w) {
    e[++tot].to=v;
    e[tot].value=w;
    e[tot].next=head[u];
    head[u]=tot;
}

void DFS(int u,int father,int deepth) {
    dep[u]=deepth;
    for(int i=head[u]; i; i=e[i].next) {
        int v=e[i].to;
        if(v==father) continue;
        fa[v][0]=u,sum[v][0]=e[i].value;
        DFS(v,u,deepth+1);
    }
}

int LCA(int u,int v) {
    int ret=0x3fffffff;
    if(u==v) return 0;
    if(dep[u]<dep[v]) swap(u,v);
    for(int i=18; i>=0; i--)
        if(dep[fa[u][i]]>=dep[v]) ret=min(ret,sum[u][i]),u=fa[u][i];

    if(u==v) return ret;
    for(int j=18; j>=0; --j)
        if(fa[u][j]!=fa[v][j]) {
            ret=min(ret,min(sum[u][j],sum[v][j]));
            u=fa[u][j],v=fa[v][j];
        }
    ret=min(ret,min(sum[u][0],sum[v][0]));
    return ret;
}

int main() {
    memset(sum,127/3,sizeof sum );
    read(n),read(m);
    for(int i=1; i<=m; ++i) read(ee[i].from),read(ee[i].to),read(ee[i].value);
    for(int i=1; i<=n; ++i) dad[i]=i;
    sort(ee+1,ee+m+1,cmp);
    int cur=0;
    for(int i=1; i<=m; ++i) {
        int u=ee[i].from,v=ee[i].to,w=ee[i].value;
        u=Find(u),v=Find(v);
        if(u==v) continue;
        dad[u]=v;
        cur++;
        Add_Edge(u,v,w),Add_Edge(v,u,w);
        //if(cur==n-1) break;
    }
    for(int i=1; i<=n; ++i)
        if(!dep[i]) DFS(i,0,1);
    for(int j=1; j<=18; ++j)
        for(int i=1; i<=n; ++i)
            if(fa[i][j-1]!=0) {
                fa[i][j]=fa[fa[i][j-1]][j-1];
                sum[i][j]=min(sum[i][j-1],sum[fa[i][j-1]][j-1]);
            }
    int T;
    read(T);
    for(int u,v; T--;) {
        read(u),read(v);
        if(Find(u)!=Find(v)) {
            printf("-1\n");
            continue;
        }
        int ans=LCA(u,v);
        if(ans==0x3fffffff) ans=-1;
        printf("%d\n",ans);
    }

    return 0;
}