银河英雄传说洛谷1196 并查集

最新推荐文章于 2025-02-11 20:43:12 发布

原创最新推荐文章于 2025-02-11 20:43:12 发布 · 383 阅读

0 ·

CC 4.0 BY-SA版权

OI刷题专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种使用并查集解决有趣的食物链问题的方法。通过维护每个节点到其祖先的距离和祖先集合中元素的数量，实现了高效的查找和更新操作。特别地，文章详细解释了如何在并查集中使用路径压缩，并提供了完整的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

　　有点有趣的并查集问题，类似于食物链而略简单

　　使用sum[i]表示以i为祖先的并查集的元素个数（不包含i），以dis[i]表示i到其祖先的距离，每次getfather时根据x的直接父亲来更新x的dis值，每次修改操作时用sum[y]更新dis[x]，用sum[x]更新sum[y]，最后输出时直接利用两个点的dis值得解

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 
 5 
 6 const int maxn = 30000 + 500;
 7 int father[maxn], dis[maxn];
 8 int sum[maxn];
 9 char c;
10 int x, y;
11 int t;
12 
13 int getfather(int x) {
14     if (father[x] == x) return (x);
15     int cur = father[x];
16     father[x] = getfather(father[x]);
17     dis[x] = dis[x] + dis[cur];
18     return (father[x]);
19 }
20 
21 int main () {
22     scanf("%d", &t);
23     for (int i = 1; i <= 30000; i++) {
24         father[i] = i;
25     }
26     for (int i = 1; i <= t; i++) {
27         scanf(" %c %d %d", &c, &x, &y);
28         if (c == 'M') {
29             int tx = getfather(x);
30             int ty = getfather(y);
31             if (tx != ty) {
32                 father[tx] = ty;
33                 dis[tx] = ++sum[ty];
34                 sum[ty] += sum[tx];
35             }
36         } else {
37             int tx = getfather(x);
38             int ty = getfather(y);
39             if (tx != ty) printf("-1\n");
40             else printf("%d\n", abs(dis[x] - dis[y]) - 1);
41         }
42     }
43 
44     return 0;
45 }