任重而道远
题目描述
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。
输入输出样例
输入样例#1： 复制
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3输出样例#1： 复制
3
-1
3
AC代码：
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int oo = 1e9 + 7;
const int N = 1e4 + 5;
const int M = 5e4 + 5;
const int P = 20;
struct Edge_0 {
int x, y, z;
bool operator < (const Edge_0& nd) const {
return z > nd.z;
}
}p[M << 1];
struct Edge {
int tov, nxt, val;
}e[N << 1];
int head[N << 1], fa[N], vis[N], dep[N], anc[N][P + 2], w[N][P + 2];
int num, n, m, q;
void add_edge (int u, int v, int val) {
num++;
e[num] = (Edge) {v, head[u], val};
head[u] = num;
}
int find (int x) {
return x == fa[x] ? x : fa[x] = find (fa[x]);
}
void Kruskal () {
int tot = 0;
for (int i = 1; i <= n; i++) fa[i] = i;
for (int i = 1; i <= m; i++) {
int u = p[i].x, v = p[i].y, x = p[i].z;
int fu = find (u), fv = find (v);
if (fu == fv) continue;
add_edge (u, v, x);
add_edge (v, u, x);
fa[fu] = fv, tot++;
if (tot == n - 1) continue;
}
}
void dfs (int u) {
vis[u] = 1;
for (int i = head[u]; i; i = e[i].nxt) {
int v = e[i].tov, val = e[i].val;
if (vis[v]) continue;
dep[v] = dep[u] + 1;
anc[v][0] = u, w[v][0] = val;
dfs (v);
}
}
void init () {
for (int i = 1; i <= n; i++)
if (!vis[i]) {
dep[i] = 1;
dfs (i);
anc[i][0] = i;
w[i][0] = oo;
}
for (int i = 1; i <= P; i++)
for (int j = 1; j <= n; j++) {
anc[j][i] = anc[anc[j][i - 1]][i - 1];
w[j][i] = min (w[j][i - 1], w[anc[j][i - 1]][i - 1]);
}
}
int lca (int u, int v) {
int sww = oo;
if (dep[u] < dep[v]) swap (u, v);
for (int p = P; p >= 0; p--)
if (dep[anc[u][p]] >= dep[v]) {
sww = min (sww, w[u][p]);
u = anc[u][p];
}
if (u == v) return sww;
for (int p = P; p >= 0; p--)
if (anc[u][p] != anc[v][p]) {
sww = min (sww, min (w[u][p], w[v][p]));
u = anc[u][p], v = anc[v][p];
}
sww = min (sww, min (w[u][0], w[v][0]));
return sww;
}
int main () {
scanf ("%d%d", &n, &m);
for (int i = 1; i <= m; i++) {
int x, y, z;
scanf ("%d%d%d", &x, &y, &z);
p[i] = (Edge_0) {x, y, z};
}
sort (p + 1, p + m + 1);
Kruskal ();
init ();
scanf ("%d", &q);
while (q--) {
int u, v;
scanf ("%d%d", &u, &v);
if (find (u) != find (v)) {
printf ("-1\n");
goto Nxt;
}
printf ("%d\n", lca (u, v));
Nxt:;
}
return 0;
}