poj 1988 Cube Stacking （poj 1182 食物链(转))

昨晚上和今一早，做了食物链后，便做了这个题，做的郁闷。刚开始的时候我拿最下面的当
根节点，做出来后发现这样会漏情况的。
比如：
11
M 1 10
M 2 10
M 3 10
M 4 10
M 5 10
M 10 6
C 10
C 4
M 4 8
C 3
C 4
这组测试数据，在M 4 8 合并后，元素3的下方就会漏掉一个箱子。

后来实在没办法了，上网看了看，大家都是以最上面的为根节点o(╯□╰)o（自己好笨。。。），
那样就不会出现问题了。

自己写完代码后，怎么交都是WA，唉，然后没办法就又看了网上的源代码，一看几乎一样，
但是我的就是交不上去，实在找不到哪里不对了，索性先把代码粘上来吧，请大家帮我找找错误吧。
下面是真确的代码。

 

它的思路大概就和食物链很像，每个元素要进行合并，在合并的时候还要处理它的一个关系，即它上面的箱子数，最后用这个集合的总数减去该箱子的上面的数目，再-1 就OK了。

 如果先做了食物链，这个就很好想了。

 

自己感觉这两道题挺不错的，加深了我对并查集的理解。

 

 

下面是我的代码，请大家帮我找找哪不对吧  谢谢！

 

 

#include "stdio.h" #include "iostream" using namespace std; int p[30002],r[30002],rank[30002]; //p[i]为i的根节点，r[i]为i上面的箱子数，rank[i]为i所在的该集合的 //总箱子数 int make() { int i; for(i=0; i<=30002; i++) { p[i]=i;r[i]=0;rank[i]=1; } } int find(int x) { if(x!=p[x]) { r[x]+=r[p[x]]; p[x]=find(p[x]); } return p[x]; } int Union(int x,int y) { x=find(x); y=find(y); p[y]=x; r[y]=rank[x]; rank[x]+=rank[y]; } int main() { int x,y,it,n,m; char c; scanf("%d",&n); make(); for(int e=0; e<n; e++) { cin>>c; if(c=='C') { scanf("%d",&x); it=find(x); printf("%d/n",rank[it]-r[x]-1); } else { scanf("%d %d",&x,&y); Union(x,y); } } return 0; }

这是AC的代码

 

#include<iostream> #include<stdio.h> using namespace std; struct node{ int up,count,parent; }; node Cu[30001]; void creat(){ int c; for(c=0;c<=30000;c++){ Cu[c].count=1; Cu[c].parent=c; Cu[c].up=0; } } int find_set(int i){ int t=Cu[i].parent; if(t!=i){ Cu[i].parent=find_set(t); Cu[i].up+=Cu[t].up; } return Cu[i].parent; } void uion(int i,int j){ int x=find_set(i); int y=find_set(j); Cu[y].parent=x; Cu[y].up=Cu[x].count; Cu[x].count+=Cu[y].count; } int main(){ creat(); int n,a,b; char s; cin>>n; while(n--){ cin>>s; if(s=='M'){ scanf("%d %d",&a,&b); uion(a,b); } else if(s=='C'){ scanf("%d",&a); b=find_set(a); printf("%d/n",Cu[b].count-Cu[a].up-1); } } system("pause"); return 0; }

 

这种并查集的解法我是看一个很N 的思路看懂的，下面这两页，大家有兴趣看看。

http://www.360doc.com/content/10/0816/19/2692170_46550108.shtml

http://www.360doc.com/content/10/0816/19/2692170_46550429.shtml

 

自己感觉这两道题挺不错的，加深了我对并查集的理解。