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;}