首页 > 代码库 > 【POJ-1988】Cube Stacking(并查集)
【POJ-1988】Cube Stacking(并查集)
//===================================== // KinderRiven POJ 1899 //===================================== #include<cstdio> #include<cstring> using namespace std; const int maxn = 33333; const int INF = 30000; int fa[maxn]; //父亲结点的编号 int ret[maxn]; //压在i下面有几个木块 int size[maxn]; //以结点i为底的栈中元素的个数,如果i不是根,那么size[i] = 0; void init(){ for(int i = 1; i <= INF; i++){ fa[i] = i; ret[i] = 0; size[i] = 1;} } int find_father(int u){ if(fa[u] != u){ int temp = fa[u]; fa[u] = find_father(fa[u]); if(size[u]){ //size[i]不为0说明该元素为栈底的元素 ret[u] += size[temp]; size[temp] += size[u]; size[u] = 0; } else //否则的话这个元素就是一个普通的元素 ret[u] += ret[temp]; } return fa[u]; } void union_set(int p,int q){ int fp = find_father(p); int fq = find_father(q); fa[fp] = fq; find_father(p); find_father(q); return; } int main(){ int n; init(); scanf("%d",&n); while(n--){ char op[10]; scanf("%s",op); if(op[0] == 'M'){ int x,y; scanf("%d%d",&x,&y); union_set(x,y); } else{ int x; scanf("%d",&x); find_father(x); printf("%d\n",ret[x]); } } return 0; }
【POJ-1988】Cube Stacking(并查集)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。