题目地址:POJ 1988
这道题的查找合并的方法都能想的到,就是一点没想到,我一直天真的以为查询的时候,输入后能立即输出,这种话在合并的时候就要所有的结点值都要算出来,可是经过路径压缩之后,没办法所有都处理到,假设不压缩妥妥的TLE。。于是看了看网上的题解。才发现自己是多么的天真(ben,四声)。。在查询的时候仅仅要找一次跟就能够了。。这样不需查询的也就不是必需处理出来。反而更省时。
这题的基本思路是非常好想的。另开两个数组,一个记录以此节点为根的子节点的数目(这样合并的时候仅仅须要加还有一个根的数目即可了),还有一个用来记录这个结点以下的点的数目,即须要输出的答案。然后分别在查找和合并的时候进行维护就能够了。
代码例如以下:
#include <iostream> #include <cstdio> #include <string> #include <cstring> #include <stdlib.h> #include <math.h> #include <ctype.h> #include <queue> #include <map> #include <set> #include <algorithm> using namespace std; int bin[31000], rank[31000], num[31000]; int find1(int x) { int y; if(bin[x]!=x) { y=bin[x]; bin[x]=find1(bin[x]); rank[x]+=rank[y]; } return bin[x]; } void join(int x, int y) { int f1=find1(x); int f2=find1(y); if(f1!=f2) { bin[f1]=f2; rank[f1]+=num[f2]; num[f2]+=num[f1]; } } int main() { int n, a, b, i, f1, f2; char c; scanf("%d",&n); for(i=1;i<=30000;i++) { bin[i]=i; rank[i]=0; num[i]=1; } while(n--) { getchar(); scanf("%c",&c); if(c=='M') { scanf("%d%d",&a,&b); join(a,b); } else { scanf("%d",&a); find1(a); printf("%d ",rank[a]); } } return 0; }