hdu 2818 Building Block (並查集)(根據父節點來更新)

題意:有兩種操作:一個是M,a,b;表示a集合所有的元素,置於b集合之上;

                                       二是C  ,a;表示a集合下的元素的個數有多少個;

此題;要構造兩個數組,一個是rank【i】,他記錄的是i集合元素的總個數;另一個是under【i】,記錄的是i元素下的個數,這個集合要更新,在並查集裏進行更新 ,他會從k一直更新到他的父節點,這樣其中的所有的點便都有了因隨時操作所引起的元素個數的變化;

在合併的時候,都合併到第二個集合上;這樣保證了這個集合的父節點正好位於最下邊,則有under【m】=rank【n】;

在操作c是一定要再次更新一次即find(m);保證結果的正確性;

 

#include"stdio.h"
#include"string.h"
int pre[100006],rank[100006];
int under[100006];
int find(int k)//合併的同時,也更新了under;
{
 int temp;
 if(k!=pre[k])
 {
  temp=find(pre[k]);
  under[k]+=under[pre[k]];
  pre[k]=find(pre[k]);
 }
 return pre[k];
}
int main()
{
 int m,n,i,k;
 char a[10];
 while(scanf("%d",&k)!=EOF)
 {
  for(i=0;i<=k;i++)
  {
   pre[i]=i;
   rank[i]=1;
   under[i]=0;
  }
  while(k--)
  {
  scanf("%s",a);
  if(a[0]=='M')
  {
   scanf("%d%d",&m,&n);
   m=find(m);
   n=find(n);
   if(m!=n)
   {
    pre[m]=n;
    under[m]=rank[n];
    rank[n]+=rank[m];
   }
  }
  else
  {
   scanf("%d",&m);
   find(m);
   printf("%d\n",under[m]);
  }
  }
 }
 return 0;
}

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章