NOIP 2013 货车运输

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

基本思路就是先跑一遍最大生成树，然后建树lca…

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int inf = 200000000;
const int SZ = 100100;

struct Edge
{
    int f, t, d;
}es[SZ], tr[SZ];
int n, m, fa[SZ], first[SZ], nxt[SZ], tot = 1, jump[SZ][20], dis[SZ][20], deep[SZ];
bool vis[SZ];

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

int find(int x)
{
    return fa[x] == x ? x : fa[x] = find(fa[x]);
}

void Build(int f, int t, int d)
{
    tr[++tot] = (Edge){f, t, d};
    nxt[tot] = first[f];
    first[f] = tot; 
}

void dfs(int u)
{
    vis[u] = 1;
    for(int i = 1; i <= 16; i++)
    {
        if(deep[u] < (1 << i)) break;
        jump[u][i] = jump[jump[u][i - 1]][i - 1];
        dis[u][i] = min(dis[u][i - 1], dis[jump[u][i - 1]][i - 1]);
    }
    for(int i = first[u]; i; i = nxt[i])
    {
        int v = tr[i].t;
        if(vis[v]) continue;
        jump[v][0] = u;
        dis[v][0] = tr[i].d;
        deep[v] = deep[u] + 1;
        dfs(v);
    }
}

int lca(int x, int y)
{
    if(deep[x] < deep[y]) swap(x, y);
    int t = deep[x] - deep[y];
    for(int i = 0; i <= 16; i++)
        if((1 << i) & t) x = jump[x][i];
    for(int i = 16; i >= 0; i--)
        if(jump[x][i] != jump[y][i])
            x = jump[x][i], y = jump[y][i];
    if(x == y) return x;
    return jump[x][0];
}

int ask(int x, int f)
{
    int minn = inf;
    int t = deep[x] - deep[f];
    for(int i = 0; i <= 16; i++)
        if(t & (1 << i))
        {
            minn = min(minn, dis[x][i]);
            x = jump[x][i];
        }
    return minn;
}

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++) fa[i] = i;
    for(int i = 1; i <= m; i++)
        scanf("%d%d%d", &es[i].f, &es[i].t, &es[i].d);
    sort(es + 1, es + m + 1, cmp);  
    int cnt = 0;
    for(int i = 1; i <= m; i++)
    {
        int x = es[i].f, y = es[i].t;
        int p = find(x), q = find(y);
        if(p != q)
        {
            fa[p] = q;
            Build(x, y, es[i].d);
            Build(y, x, es[i].d);
            cnt++; 
            if(cnt == n - 1) break;
        }
    }
    for(int i = 1; i <= n; i++)
        if(!vis[i]) dfs(i);
    int q, x, y;
    scanf("%d", &q);
    for(int i = 1; i <= q; i++)
    {
        scanf("%d%d", &x, &y);
        if(find(x) != find(y)) printf("-1\n");
        else
        {
            int t = lca(x, y);
            printf("%d\n", min(ask(x, t), ask(y, t)));
        }
    }
    return 0; 
}