题目介绍
团伙问题(并查集)
在城市里住着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;
}
}
}
运行截图
记录点滴,乐于分享,转载请注明出处。
愿我们能以梦为马,不负人生韶华!
与君共勉!