zoj - 2833 - Friendship
这是一道简单的并查集题目,对于每个人,都先建一棵以他为根的根树,用数组实现树的思想,只要一个fa[]数组就够了,
然后写一个寻找树根的函数,因为我们每次操作都是对树根进行的,再写一个判断函数,判断在同一棵树中的时候,
返回1,不在同一棵树的时候,把层数小的并到层数大的树中,返回0,另外一个数组,保存以每个人为根的朋友数。
就这样了,TLE!(不听老人言,用cin没用scanf) CE!(点了C提交) AC!(换回C++提交)
#include <stdio.h> #include <string.h> const int maxn = 100000 + 10; //最多可能有100000个人 int fa[maxn], cnt[maxn], height[maxn]; //fa[i]为i的朋友,就是树的结点的父亲,cnt[i]为以i为根的朋友数,height[i]为i为根的树的层数 int find(int x) //寻找x的根,朋友中的老大!!! { while(fa[x] != x) { x = fa[x]; } return x; } int judge(int x, int y) //判断x与y的关系 { int xx = find(x); int yy = find(y); if(xx == yy) //如果x与y已在同一棵树中,返回1 return 1; else //如果x与y不在同一棵树中 { if(height[xx] > height[yy]) //当x所在的树的高度>y所在的树的高度时 { fa[yy] = xx; cnt[xx] += cnt[yy]; } else if(height[xx] == height[yy]) //当x所在的树的高度==y所在的树的高度时 { fa[yy] = xx; cnt[xx] += cnt[yy]; height[xx]++; } else //当x所在的树的高度<y所在的树的高度时 { fa[xx] = yy; cnt[yy] += cnt[xx]; } return 0; } } int main() { int N, M, i, a, b, count = 1, first = 1; //N个人,M行数据,输入变量a,b,count为测试组数,first为输出标记(判断是否要空行) char c; while(scanf("%d%d", &N, &M) != EOF) //C的写法! { if(first) first = !first; //控制第一次不要输出空行,以后每组测试数据都输出一个空行 else printf("\n"); printf("Case %d:\n", count++); //可能觉得除第1组测试数据外,以后的数据输出这行的位置有点别扭,不过,不影响AC! for(i = 1; i <= N; i++) //初始化 { fa[i] = i; //开始每个人都是一棵根树 height[i] = 1; //就只有他自己一层 cnt[i] = 1; //就只有他自己一个人,孤独!寂寞!空虚! } getchar(); //吸收标记,不起影响力数据,但没有不行! for(i = 0; i < M; i++) { scanf("%c", &c); if(c == 'M') //当判断为关系的时候 { scanf("%d%d", &a, &b); judge(a, b); getchar(); //吸收标记,不起影响力数据,但没有不行! } else if(c == 'Q') //当判断为询问的时候 { scanf("%d", &a); printf("%d\n", cnt[find(a)]); getchar(); //吸收标记,不起影响力数据,但没有不行! } } } return 0; }