【小米笔试题】朋友圈问题-并查集解法

来源：互联网发布：从零开始学java 编辑：程序博客网时间：2024/06/06 14:09

2016年小米校招笔试第三题（西安站）

3 假如已知有n个人和m对好友关系（存于数字r）。如果两个人是直接或间接的好友（好友的好友的好友...），则认为他们属于同一个朋友圈，请写程序求出这n个人里一共有多少个朋友圈。
假如：n = 5，m = 3，r = {{1 , 2} , {2 , 3} , {4 , 5}}，表示有5个人，1和2是好友，2和3是好友，4和5是好友，则1、2、3属于一个朋友圈，4、5属于另一个朋友圈，结果为2个朋友圈。

参考解法（c/c++版）：<Q：316190672，欢迎交流>

// 简单的并查集应用#include<stdio.h>#include<stdlib.h>int set[5];inline int find(int x)           //带路径优化的并查集查找算法{int i, j, r;r = x;while (set[r] != r)r = set[r];i = x;while (i != r){j = set[i];set[i] = r;i = j;}return r;}inline void merge(int x, int y)     //优化的并查集归并算法{int t = find(x);int h = find(y);if (t < h)set[h] = t;elseset[t] = h;}int friends(int n, int m, int r[11][2]){int i, count;for (i = 1; i <= n; ++i)    //初始化并查集，各点为孤立点，分支数为n set[i] = i;for (i = 0; i < m; ++i)merge(r[i][0], r[i][1]);count = 0;for (i = 1; i <= n; ++i){if (set[i] == i)++count;}return count;}int main(void){/* int r[11][2] = { { 1, 5 }, { 3, 5 }, { 4, 5 }, { 1, 4 }, { 5, 6 }, { 8, 1 }, { 9, 20 }, { 98, 11 }, { 13, 76 }, { 98, 77 }, { 2, 1 } };*/int r[5][2] = { { 1, 2 }, { 2, 3 }, {4,5}};int i;for (i = 0; i < sizeof(r[5][2]) / sizeof(r[0]);i++){merge(r[i][0],r[i][1]);}printf("%d\n",friends(5, 3, r));system("PAUSE");return 0;}

运行结果：

0 0