题目大意:
(来源于:洛谷)
解题思路:
- 最大生成树去掉没用的边
- 利用倍增的方法存储每个结点向上的最小值
- 最后利用倍增Lca,在算Lca的过程中,顺便计算沿途最小值
说明:
最大生成树,只保留了权值最大的那些边,而且每个点互相之间都可以到达,不用经过那些权值较小的边
注意:
- Lca倍增时,要先算最小值,再倍增
- Lca出现祖先结点为0时要退出(数组不要越界)
- 记住先判断是否属于同一颗树
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e4 + 10;
const int maxm = 5e4 + 10;
int n, m, q;
int tot, head[maxn];
struct Tree {
int nxt, to, w;
} edge[maxm];
void add_edge(int u, int v, int w) {//存储最大生成树
edge[++tot].nxt = head[u], head[u] = tot, edge[tot].to = v, edge[tot].w = w;
edge[++tot].nxt = head[v], head[v] = tot, edge[tot].to = u, edge[tot].w = w;
}
int k = 25, f[maxn][30], f_min[maxn][30], d[maxn];
//f代表祖先,f_min代表往上的最小值
void dfs(int u, int pre) {
for (int i = 1; i <= k; i++) {
f[u][i] = f[f[u][i - 1]][i - 1], f_min[u][i] = min(f_min[u][i - 1], f_min[f[u][i - 1]][i - 1]);
if (!f[u][i]) break;
}
for (int i = head[u]; i; i = edge[i].nxt) {
int v = edge[i].to, w = edge[i].w;
if (v == pre) continue;
d[v] = d[u] + 1;
f[v][0] = u, f_min[v][0] = w;//初始化
dfs(v, u);
}
}
int fa[maxn], num[maxn];//fa代表同属哪颗树,num代表所属的树有几个结点
struct Graph {
int u, v, w;
} G[maxm];
bool cmp(const Graph A, const Graph & B) {
return A.w > B.w;
}
int find(int u) {
if (u == fa[u]) return u;
else return find(fa[u]);
}
void combine(int u, int v) {
if (num[u] < num[v]) swap(u, v);
int root_u = find(u), root_v = find(v);
num[root_u] += num[root_v];
while (fa[v] != v) {
int tmp = v;
v = fa[v], fa[tmp] = root_u;
}
fa[v] = root_u;
}
void kruskal() {
for (int i = 1; i <= m; i++) {
if (find(G[i].u) != find(G[i].v)) {
add_edge(G[i].u, G[i].v, G[i].w);
combine(G[i].u, G[i].v);
}
}
}
int Lca(int u, int v) {
int ans = 1e5 + 10;
if (d[u] < d[v]) swap(u, v);
for (int i = k; i >= 0; i--) {
if (d[f[u][i]] > d[v])
ans = min(ans, f_min[u][i]), u = f[u][i];
}
if (d[u] != d[v]) ans = min(ans, f_min[u][0]), u = f[u][0];
for (int i = k; i >= 0; i--) {
if (f[u][i] != f[v][i]) {
ans = min(ans, min(f_min[u][i], f_min[v][i]));
u = f[u][i], v = f[v][i];
}
}
if (u != v)
ans = min(ans, min(f_min[u][0], f_min[v][0])), u = f[u][0];
return ans;
}
int main() {
int u, v, w;
cin >> n >> m;
for (int i = 1; i <= m; i++)
cin >> G[i].u >> G[i].v >> G[i].w;
for (int i = 1; i <= n; i++) fa[i] = i, num[i] = 1;
sort(G + 1, G + m + 1, cmp);
kruskal();
dfs(1, 0);
cin >> q;
while (q--) {
cin >> u >> v;
if (find(u) != find(v)) {//先判断是不是同一棵树
cout << -1 << endl;
continue;
}
cout << Lca(u, v) << endl;
}
}