题目: POJ 1988 Cube Stacking ,哈哈,我们今天来看一道复杂一点的题嘛,这是选自POJ上的一道题,好了,我们一起来看看题意吧:
考虑到直接复制题目,或者截屏的方式不是很方便阅读,我就把直接题目链接放下面!
题目传送门:http://poj.org/problem?id=1988
思路
:
读了题目,思考了很久,才发现这道题就是个带权重的并查集,具体就看代码吧!
我们来看看成功AC的代码吧:
#include<iostream> #include<string> using namespace std; int n; int p[40000]; int cnt[40000];//记录当前节点到根节点的距离(不是树中的距离,而是在栈中,) int total[40000];//记录树的总节点个数 int find(int x){ if(x!=p[x]){ int t=p[x]; p[x]=find(p[x]); cnt[x]+=cnt[t]; } return p[x]; } void hb(int x,int y){ int xx=find(x),yy=find(y); p[xx]=yy; cnt[xx]+=total[yy]; total[yy]+=total[xx]; } int main(){ cin>>n; for(int i=0;i<40000;i++){ p[i]=i; cnt[i]=0; total[i]=1; } while(n--){ string s; cin>>s; if(s=="M"){ int x,y; cin>>x>>y; hb(x,y); }else{ int x; cin>>x; find(x); cout<<cnt[x]<<"\n"; } } return 0; }