当前位置: 代码迷 >> 综合 >> zoj - 2833 - Friendship
  详细解决方案

zoj - 2833 - Friendship

热度:94   发布时间:2024-01-10 14:03:11.0

这是一道简单的并查集题目,对于每个人,都先建一棵以他为根的根树,用数组实现树的思想,只要一个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已在同一棵树中,返回1return 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;
}