洛谷 P5092【加权并查集】

最新推荐文章于 2025-08-05 16:33:16 发布

willem248

最新推荐文章于 2025-08-05 16:33:16 发布

阅读量372

点赞数

CC 4.0 BY-SA版权

分类专栏：洛谷文章标签： c++ 算法数据结构

转载请注明出处，谢谢。

本文链接：https://blog.csdn.net/lxylluvio/article/details/126730403

洛谷专栏收录该内容

32 篇文章

订阅专栏

这是一个关于BCT创客大赛的程序代码，实现了并查集的数据结构，用于处理合并与查询操作。代码中包含初始化、查找、合并及查询功能，适用于解决集合合并与元素所属集合查询的问题。

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

话说这不是bct创客大赛普及组的第三题吗

#include <bits/stdc++.h>
#define int long long

using namespace std;

int read() {
  int x;
  scanf ("%lld", &x);
  return x;
  /*
  char ch;
  int opt = 1, num;
  while (!isdigit(ch = getchar())) {
    if (ch == '-')
      opt = -1;
  }
  num = ch ^ 48;
  while (isdigit(ch = getchar()))
    num = (num << 3LL) + (num << 1LL) + (ch ^ 48);
  return num * opt;
  */
}

const int N = 1e6 + 233;
int fa[N], w[N], fr[N];

int find(int x) {
  if (fa[x] == x)
    return x;
  int tx = find(fa[x]);
  fr[x] += fr[fa[x]];
  return fa[x] = tx;
}

void merge(int x, int y) {
  swap (x, y);
  int tx = find(x);
  int ty = find(y);
  fr[tx] += w[ty];
  fa[tx] = ty;
  w[ty] += w[tx];
}

void init(int n) {
  for (int i = 1; i <= n; i ++)
    fa[i] = i;
  for (int i = 1; i <= n; i ++)
    w[i] = 1;
  for (int i = 1; i <= n; i ++)
    fr[i] = 0;
}

int query(int x) {
  // return w[x];
  int ta = find(x);
  return w[ta] - fr[x] - 1;
}

signed main() {
  int n = read();
  init(33333);
  while (n --) {
    char ch;
    cin >> ch;
    if (ch == 'M') {
      int a, b;
      a = read(), b = read();
      merge(a, b);
    } else {
      int x;
      x = read();
      printf ("%lld\n", query(x));
    }
  }
  return 0;
}