小韦老师@神犇营-my0411-团伙

小韦老师@神犇营-my0411-团伙

题目：

描述

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

1、我朋友的朋友是我的朋友；

2、我敌人的敌人是我的朋友；

所有是朋友的人组成一个团伙。告诉你关于这 n 个人的 m 条信息，即某两个人是朋友，或者某两个人是敌人，请你编写一个程序，计算出这个城市最多可能有多少个团伙？

输入

第 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 是敌人。

输出

一个整数，表示这 n 个人最多可能有几个团伙。

输入样例1

6 4
1 1 4
0 3 5
0 4 6
1 1 2

输出样例1

3

题解：

思路

整体思路：

用 a 表示 a 的朋友；

a + n 表示 a 的敌人。

如果 a 和 b 是朋友，合并 a 所在集合和 b 所在集合；

如果 a 和 b 是敌人，合并 a + n 所在集合和 b 所在集合，b + n 所在集合和 a 所在集合。

具体步骤：

1. 定义两个数组：

    // 数组要开成人数的两倍加 10，因为要把 i + n 当作 i 的敌人 
    const int N = 2e3 + 10;  
    int father[N];  // 父亲数组 
    bool isRoot[N];  // 根数组，一开始都初始化为 false 

2. 调用初始化函数进行初始化：

	init(n);

3. 输入 m 组数据：

    for (int i = 0; i < m; i++) {
    	cin >> p >> x >> y;
    	if (p == 0) {  // 若 x 和 y 是朋友，则合并 x 和 y 
    		Union(x, y);
    	} else {  // 若 x 和 y 是敌人 
    		Union(x + n, y);  // 合并 x 的敌人 
    		Union(x, y + n);  // 合并 y 的敌人 
    	}
    } 

4. 数有多少个根，即有多少个集合（有多少个团伙）：

	int cnt = 0;  // 计数器 
	for (int i = 1; i <= n; i++) {  // 枚举 1~n 
		int t = findFather(i);  // 找到 i 的根 
		if (!isRoot[t]) {  // 若该根未被标记（每个根只能数一次） 
			isRoot[t] = true;  // 置为 true，说明已标记 
			cnt++;  // 计数器加 1 
		}
	}

5. 初始化函数 init 的实现：

	// 初始化数组，1~n 是自己，n+1 ~ n+n 是 1~n 的敌人 
	void init(int n) {
		for (int i = 1; i <= 2*n; i++) {
			father[i] = i;
		}
	}

6. 合并函数的实现：

	// 合并 x 所在集合和 y 所在集合 
	void Union(int x, int y) {
		int faX = findFather(x);
		int faY = findFather(y);
		if (faX != faY) {
			father[faY] = faX;
		}
	}

7. 查找集合的根的函数的实现：

	// 查找 x 所在的集合的根 
	int findFather(int x) {
		int a = x;
		while (x != father[x]) {
			x = father[x];
		} 
		// 压缩路径（在时间要求不太严格时可以不写） 
		while (a != father[a]) {
			int z = a;
			a = father[a];
			father[z] = x;
		}
		return x;  // 返回根 
	}

思考：

1° 为什么第 4 步要那样去数集合的根，换个方式行吗？例如数有多少个 father[i] == i ？

2° 该题解中对于“敌人的敌人是朋友”的处理方式给你什么样的启发？

完整代码：

#include <bits/stdc++.h>

using namespace std;

// 数组要开成人数的两倍加 10，因为要把 i + n 当作 i 的敌人 
const int N = 2e3 + 10;  
int father[N];  // 父亲数组 
bool isRoot[N];  // 根数组，一开始都初始化为 false 
int n, m;

// 初始化数组，1~n 是自己，n+1 ~ n+n 是 1~n 的敌人 
void init(int n) {
	for (int i = 1; i <= 2*n; i++) {
		father[i] = i;
	}
}

// 查找 x 所在的集合的根 
int findFather(int x) {
	int a = x;
	while (x != father[x]) {
		x = father[x];
	} 
	// 压缩路径（在时间要求不太严格时可以不写） 
	while (a != father[a]) {
		int z = a;
		a = father[a];
		father[z] = x;
	}
	return x;  // 返回根 
}

// 合并 x 所在集合和 y 所在集合 
void Union(int x, int y) {
	int faX = findFather(x);
	int faY = findFather(y);
	if (faX != faY) {
		father[faY] = faX;
	}
} 
 
int main() {
	
	cin >> n >> m;
	init(n);  // 记得要调用初始化函数 
	int p, x, y;
	for (int i = 0; i < m; i++) {
		cin >> p >> x >> y;
		if (p == 0) {  // 若 x 和 y 是朋友，则合并 x 和 y 
			Union(x, y);
		} else {  // 若 x 和 y 是敌人 
			Union(x + n, y);  // 合并 x 的敌人 
			Union(x, y + n);  // 合并 y 的敌人 
		}
	}
	int cnt = 0;  // 计数器 
	for (int i = 1; i <= n; i++) {  // 枚举 1~n 
		int t = findFather(i);  // 找到 i 的根 
		if (!isRoot[t]) {  // 若该根未被标记（每个根只能数一次） 
			isRoot[t] = true;  // 置为 true，说明已标记 
			cnt++;  // 计数器加 1 
		}
	}
	cout << cnt;  // 输出结果 

	return 0;
}

我是小韦老师，企者不立，跨者不行，每天进步一点点。

欢迎大家多多交流，如果发现有错误，请多指正。有疑问的同学也可以留言评论或者发邮件。邮箱：weichangying_wcy@163.com