题目介绍
团伙问题(并查集)
在城市里住着n个人,任何两个认识的人不是朋友就是敌人,而且满足:
(1)朋友的朋友是朋友;
(2)敌人的敌人是朋友。
这n个人可以划分为若干个团伙,使得每个团伙中任意两个成员均为朋友。告诉你关于这n个人的m条信息,即某两个人是朋友,或者某两个人是敌人,请你编写一个程序,计算这个城市最多可能有多少个团伙?
Input
输入包含多组数据,对于每组数据:
第1行为n和m,1
以下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; } } }
运行截图