給定一個可能不連通的無向加權圖,需回答多組查詢:若兩城市不連通輸出"Not connected",否則輸出兩點間的最短距離。節點數N≤10000,邊數M未指定但隱含較大。
dis[u] + dis[v] - 2*dis[lca]計算距離。
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 10010;
struct Edge { int to, cost; };
vector<Edge> G[MAXN];
int dp[MAXN*2][20], L[MAXN*2], E[MAXN*2], H[MAXN], dis[MAXN];
bool vis[MAXN];
int fa[MAXN], n, k, T;
int find_set(int x) { return fa[x]==x ? x : fa[x]=find_set(fa[x]); }
void unite(int a, int b) { fa[find_set(a)] = find_set(b); }
void dfs(int u, int dep) {
vis[u] = true;
E[++k] = u; L[k] = dep; H[u] = k;
for (auto &e : G[u]) if (!vis[e.to]) {
dis[e.to] = dis[u] + e.cost;
dfs(e.to, dep+1);
E[++k] = u; L[k] = dep;
}
}
void rmq_init(int len) {
for (int i=1; i<=len; ++i) dp[i][0] = i;
for (int j=1; (1<<j)<=len; ++j)
for (int i=1; i+(1<<j)-1<=len; ++i)
dp[i][j] = (L[dp[i][j-1]] < L[dp[i+(1<<(j-1))][j-1]]) ? dp[i][j-1] : dp[i+(1<<(j-1))][j-1];
}
int lca(int u, int v) {
u = H[u]; v = H[v];
if (u > v) swap(u, v);
int k = 31 - __builtin_clz(v - u + 1);
return (L[dp[u][k]] < L[dp[v-(1<<k)+1][k]]) ? E[dp[u][k]] : E[dp[v-(1<<k)+1][k]];
}
int main() {
int m, a, b, c;
while (scanf("%d%d%d", &n, &m, &T) == 3) {
// 初始化
for (int i=0; i<=n; ++i) fa[i]=i, G[i].clear();
memset(dis, 0, sizeof(dis));
memset(vis, 0, sizeof(vis));
// 讀邊、建圖、合併並查集
for (int i=0; i
總結
本題巧妙結合並查集與LCA+RMQ,將不連通圖轉化為單樹問題,實現高效距離查詢。核心在於虛擬節點的引入與ST表的正確構建。