并查集--城市团伙

一．题目描述：

在某个城市里住着n个人，任何两个认识的人不是朋友就是敌人，而且满足：

    l我朋友的朋友是我的朋友；

    l我敌人的敌人是我的朋友；

    l已知关于n个人的m条信息（即某2个人是朋友或者敌人），假设所有是朋友的人一定属于同一个团伙，请计算该城市最多有多少团伙？

二．题目解析：

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

三．相关代码：

第1行为n和m，1<n<1000,1≤m≤100 000；以下m行，每行为p x y，p的值为0或1，p为0时，表示x和y是朋友，p为1时，表示x和y是敌人。

输出结果为最多团伙数。

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;

}

}

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;

}

}

}

四．结果展示：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-gGHuLrwl-1658403978957)(C:\Users\USER\AppData\Roaming\Typora\typora-user-images\image-20220721194421037.png)]
!= root2) {

persons[root2] = root1;

return root1;

} else {

return 0;

}

}

}

四．结果展示：