AcWing 837. 连通块中点的数量

Posted 2022-02-11 MangataTS

题目连接

https://www.acwing.com/problem/content/839/

思路

因为有集合关系，所以我们考虑用并查集来维护，此题对于和并和查询是否在一个集合维护比较简单，对于一个集合内的元素个数，我们可以初始化一个集合元素个数，都是1，然后每次合并的时候判断是否有两个不同集合合并，如果有的话，就把我们当作祖先集合的元素个数更新（也就是加上另外一个集合的元素个数）

代码

#include<bits/stdc++.h>
using namespace std;
//----------------自定义部分----------------
#define ll long long
#define mod 1000000009
#define endl "\\n"
#define PII pair<int,int>

int dx[4]=0,-1,0,1,dy[4]=-1,0,1,0;

ll ksm(ll a,ll b) 
	ll ans = 1;
	for(;b;b>>=1LL) 
		if(b & 1) ans = ans * a % mod;
		a = a * a % mod;
	
	return ans;


ll lowbit(ll x)return -x & x;

const int N = 2e6+10;
//----------------自定义部分----------------
int n,m,q,a[N];
int fa[N],vis[N];
void init()
	for(int i = 1;i <= n; ++i) fa[i] = i,vis[i] = 1;

int find(int x)
	int t = x;
	while(t != fa[t]) t = fa[t];
	while(x != fa[x]) 
		int temp = fa[x];
		fa[x] = t;
		x = temp;
	
	return x;


int main()

	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	std::cout.tie(nullptr);
	cin>>n>>m;
	init();
	string op;
	int u,v;
	for(int i = 0;i < m; ++i) 
		cin>>op>>u;
		u = find(u);
		if(op == "C")
			cin>>v;
			v = find(v);
			if(u != v)
				fa[v] = u,vis[u] += vis[v];
		
		else if(op == "Q1")
			cin>>v;
			v = find(v);
			cout<<(u==v?"Yes":"No")<<endl;
		
		else if(op == "Q2")
			cout<<vis[u]<<endl;
		
		
	
	
	return 0;