【短学期算法作业】团伙问题（并查集）

题目介绍

团伙问题（并查集）
在城市里住着n个人，任何两个认识的人不是朋友就是敌人，而且满足：
(1)朋友的朋友是朋友；
(2)敌人的敌人是朋友。
这n个人可以划分为若干个团伙，使得每个团伙中任意两个成员均为朋友。告诉你关于这n个人的m条信息，即某两个人是朋友，或者某两个人是敌人，请你编写一个程序，计算这个城市最多可能有多少个团伙？
Input
输入包含多组数据，对于每组数据：
第1行为n和m，1<n<1000,1<=m<=100000；
以下m行，每行为p、x、y，p的值为0或1，p为0时，表示x和y是朋友，p为1时，表示x和y是敌人。
Output：
对于每组输入数据，输出一个整数，表示这n个人最多可能有几个团伙。
Sample Input
6 4
1 1 4
0 3 5
0 4 6
1 1 2
Sample Output
4

题目思路

利用并查集来维护团伙关系，初始是每个人都是一个团伙，然后遍历输入的每一条关系并处理，如果是朋友则两个团伙合并并做路径压缩，同时团伙数减一。

具体代码

package com.dreamchaser;import java.util.Scanner;/*** 团伙问题（并查集）*/
public class test6 {/*** 记录团伙关系*/static int[] persons;static int number;/*** 人数*/static int n;/*** 关系数*/static int m;public static void main(String[] args) {Scanner scanner = new Scanner(System.in);n = scanner.nextInt();m = scanner.nextInt();persons=new int[n+1];number=n;int p,x,y;for (int i = 0; i < m; i++) {p=scanner.nextInt();x=scanner.nextInt();y=scanner.nextInt();if (p==0){if (merge(x,y)!=0){number--;}}}System.out.println(number);}/*** 寻找根节点** @param i* @return*/private static int findUnion(int i) {/*** 合并集的根*/if (persons[i] != 0) {int root = findUnion(persons[i]);//路径压缩persons[i] = root;return root;} else {return i;}}/*** 合并集合* 如果两个属于同一个集合则返回null，否则返回根节点* @param x* @param y* @return*/private static int merge(int x, int y) {int root1 = findUnion(x);int root2 = findUnion(y);if (root1 != root2) {persons[root2] = root1;return root1;} else {return 0;}}
}

运行截图

记录点滴，乐于分享，转载请注明出处。

愿我们能以梦为马，不负人生韶华！
与君共勉！