算法一是TLE的,算法二注意合并时先更新siz[],再合并
算法1
(暴力枚举) $O(n^2)$ 注意是过不了的,是我在没看之前自己想暴力看看数据能不能水过
#include<iostream>
#include<algorithm>
using namespace std;
int const maxn=100010;
int p[maxn];
int find(int x)
{
if(p[x]!=x)
p[x]=find(p[x]);
return p[x];
}
int main(void)
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i;
while(m--)
{
string op;
cin>>op;
int a,b;
if(op[0]=='C')
{
cin>>a>>b;
p[find(a)]=find(b);
}
else if(op=="Q1")
{
cin>>a>>b;
if(find(a)==find(b))
cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
else if(op=="Q2")
{
cin>>a;
int res=0;
for(int i=1;i<=n;i++) //这个循环包含了a其本身 所以res=0;
{
if(find(a)==find(i))
res++;
}
cout<<res<<endl;
}
}
return 0;
}
----------
算法2
C++ 代码
#include<iostream>
#include<algorithm>
using namespace std;
int const maxn=100010;
int p[maxn],siz[maxn];/////siz[]存的是每个集合有多少元素,但我们定义根节点是有意义的
int find(int x)
{
if(p[x]!=x)
p[x]=find(p[x]);
return p[x];
}
int main(void)
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i,siz[i]=1;
while(m--)
{
string op;
cin>>op;
int a,b;
if(op[0]=='C')
{
cin>>a>>b;
if(find(a)!=find(b)) //如果两个点的根节点不一样
{
siz[find(b)]+=siz[find(a)]; ///注意是先更新siz,不然会导致算的变少了
p[find(a)]=find(b);
}
}
else if(op=="Q1")
{
cin>>a>>b;
if(find(a)==find(b))
cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
else if(op=="Q2")
{
cin>>a;
cout<<siz[find(a)]<<endl;
}
}
return 0;
}