算法基础之连通块中点的数量

阳光男孩01

于 2023-11-27 17:34:29 发布

阅读量322

点赞数 8

文章标签：算法 c++ 数据结构图论

本文链接：https://blog.csdn.net/Pisasama/article/details/134650751

版权

连通块中点的数量

核心思想：并查集

与合并集合类似多一个cnt记录节点个数

  #include <iostream>
  
  using namespace std;
  
  const int N = 100010;
  
  
  int n, m;
  int p[N], cnt[N];
  
  int find(int x)
  {
      if (p[x] != x) p[x] = find(p[x]);
      return p[x];
  }
  
  int main(){
      cin>>n>>m;
      for (int i = 1; i <= n; i ++ )
      {
          p[i] = i;
          cnt[i] = 1;
      }
      while(m--){
          string op;
          int a, b;
          cin >> op;
          if (op == "C")
          {
              cin >> a >> b;
              a = find(a), b = find(b);
              if (a != b)
              {
                  p[a] = b;
                  cnt[b] += cnt[a];
              }
          }
          else if (op == "Q1")
          {
              cin >> a >> b;
              if (find(a) == find(b)) puts("Yes");
              else puts("No");
          }
          else
          {
              cin >> a;
              cout << cnt[find(a)] << endl;
          }
      }
      return 0;
  }