這題在去年的做並查集專題的時候,就看過當時不理解題意,一直放到現在才解決,真心蒟蒻,ORZ,,,弱爆了,最近狀態不是很好,所以要調整心態;
本題是在並查集尋找時,回溯更新under[]數組的值;用到三個數組,fa[x],記錄x的父節點,經過路徑壓縮後fa[x]的值就是x所屬的集合;con[x],表示x所在堆的個數,under[x],表示x下面節點數;很好的並查集題目;
#include<cstdio>#include<algorithm>#include<cmath>#include<queue>#include<vector>#include<string.h>#include<iostream>using namespace std;const int N=30003;int fa[N],con[N],under[N];int find(int x){ int t; if(x!=fa[x]) { t=find(fa[x]);under[x]+=under[fa[x]];fa[x]=t; } return fa[x];}void init(){ for(int i=0;i<N;i++) { fa[i]=i;con[i]=1;under[i]=0; }}void Union(int x,int y){ int xx=find(x),yy=find(y); if(xx!=yy) { under[xx]=con[yy]; con[yy]+=con[xx]; fa[xx]=yy; }}int main(){ int p,a,b; char op[11]; while(~scanf("%d",&p)){ init(); while(p--) { scanf("%s",op);if(op[0]=='M'){scanf("%d%d",&a,&b);Union(a,b);}else{scanf("%d",&a); find(a);//此處必須有,不然會wa的,要及時更新printf("%d\n",under[a]);} }} return 0;}