ps:谢谢l学长帮忙讲解这道题,以及参考的这位博主的文章,真的写的很好
分析:
除了考察并查集的基本操作以外,这里的删除操作以及集合数的返回和集合大小都需要注意一下技巧
操作一"query":
很明显是返回集合数,那么如何返回集合数?显然只需要开一个全局变量ans来维护,初始让他为n,然后随着每次操作,当集合数增加时ans++,减少时ans–,这样的话每次query只需要直接打印ans的值就可以,很方便。
操作二“fight”:
考察的是使用并查集对集合进行合并,但这里要注意一下,题目说了,人数多的赢且赢者吞并弱者,也就是说把集合大小更小的接到更大的那个上面,让更大的集合的根节点成为更小集合根节点的根节点,因此,要注意同时维护一下ans以及集合大小,所以为了更方便维护集合大小,因此这里开了一个num数组来实现;初始时因为每个节点都是自己的根节点所以num[i]都是1,这个初始化放在Init()函数里和root一起进行初始化操作;且,因为小朋友编号如果固定的话那么每次进行合并和删除操作时在怎么把要删除的节点给独立出来使得删除节点后不会影响到它的子节点就成了一个很大的问题,所以我们这里开了一个id数组专门用来存储节点编号,当你要删除某个节点x时,就让它的根节点指向一个不存在的地方,不会影响到你现在的其他节点,即令他指向n以后的点,这样就不会影响到n范围内的节点的了,所以才需要让数组开大一点,可以开成3n,方便给出足够的空间让它往后指(事实上开成2n也是可以过的,亲身实验);解决了这个问题以后然后又因为我们有num数组,这样只要这道id[x]和id[y](存储的是x和y的编号)的根节点,然后通过num直接得出根节点所在集合的大小然后根据题目要求比较后打印即可
操作三“tempt”:
这个操作要稍微注意一下细节处,因为只是把y一个人拉入x的集合,即把y并入x,如果y所在集合只有他一个人,那么y被合并以后集合数必然减少了,ans–,如果不止他一个人,那么集合数是没有变的
操作四“revolt”:
其实就是考察删除操作,不过要注意一下当集合里只有他一个人的时候集合数ans–,否则不减,所以操作三的这个过程可以通过删除+合并操作实现,就不用每个操作都写一次,写成函数的形式,就可以直接调用了
综上:通过以上分析,发现有很多步骤是有交叉的,总结起来就是四个函数的操作:初始化Init(),查找根节点Find(),合并Union(),删除del()
接下来结合代码分析:
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int root[2*N],num[2*N],id[2*N];
int cnt,ans;
void Init(int n)
{
cnt=n,ans=n;//cnt初始化为n之后++方便往n后面指
for(int i=1;i<=2*n;i++){
root[i]=i;
num[i]=1;
id[i]=i;
}
}
int Find(int x)
{
return x==root[x]?x:Find(root[x]);
}
void Union(int x,int y)
{
x=Find(x),y=Find(y);
if(x!=y){
root[y]=x;
num[x]+=num[y];//合并的同时维护集合数
ans--;//直接通过Union在合并的时候同时维护ans
}
}
void del(int x)
{
int pre=Find(id[x]);
num[pre]--;
if(num[pre]==0)ans--;
cnt++;
num[cnt]=1;
root[cnt]=cnt;//删除的时候把他的根节点指向不会影响到其他节点的地方
id[x]=cnt;
ans++;//原集合里不止一个人,从中删除一个x以后集合数增加,否则不变,所以上面会ans--使得集合数不变
}
int main()
{
int t,n,m,x,y;
scanf("%d",&t);
for(int T=1;T<=t;T++){
scanf("%d%d",&n,&m);
char s[10];
Init(n);
printf("Case #%d:\n",T);
for(int i=0;i<m;i++){
scanf("%s",s);
getchar();
if(s[0]=='q'){
printf("%d\n",ans);
}else if(s[0]=='t'){
scanf("%d%d",&x,&y);
del(y);
Union(id[x],id[y]);
}else if(s[0]=='f'){
scanf("%d%d",&x,&y);
int xx=Find(id[x]),yy=Find(id[y]);//因为是用id数组维护的编号,所以要记得找根节点的时候也是通过id,而不能直接通过x,y
if(xx==yy)continue;
if(num[xx]>num[yy])printf("%d is winner!\n",x),Union(id[x],id[y]);
else if(num[xx]<num[yy])printf("%d is winner!\n",y),Union(id[y],id[x]);
else printf("Either is winner!\n");
}else if(s[0]=='r'){
scanf("%d",&x);
del(x);
}
}
}
return 0;
}