codevs 2597 团伙

目录

1. AC 代码

题目描述 Description

1920年的芝加哥，出现了一群强盗。如果两个强盗遇上了，那么他们要么是朋友，要么是敌人。而且有一点是肯定的，就是：

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

我敌人的敌人也是我的朋友。

两个强盗是同一团伙的条件是当且仅当他们是朋友。现在给你一些关于强盗们的信息，问你最多有多少个强盗团伙。

输入描述 Input Description

输入文件gangs.in的第一行是一个整数N(2<=N<=1000)，表示强盗的个数（从1编号到N）。 第二行M(1<=M<=5000)，表示关于强盗的信息条数。 以下M行，每行可能是F p q或是E p q（1<=p q<=N），F表示p和q是朋友，E表示p和q是敌人。输入数据保证不会产生信息的矛盾。

输出描述 Output Description

输出文件gangs.out只有一行，表示最大可能的团伙数。

样例输入 Sample Input

6
4
E 1 4
F 3 5
F 4 6
E 1 2

样例输出 Sample Output

3

数据范围及提示 Data Size & Hint

2<=N<=1000

1<=M<=5000

1<=p q<=N

AC 代码

#include <iostream>
#include <vector>
using namespace std;


int fa[1003];
int flag[1003];
vector<int> e[1003];
int getfa(int x) {
	return x == fa[x]? x: fa[x] = getfa(fa[x]);
}

void merge(int x, int y) {
	fa[getfa(x)] = getfa(y);
}


int main() {
	ios::sync_with_stdio(false);
	
	int n, m;
	cin >> n;
	for (int i = 1; i <= n; i++) fa[i] = i;
	cin >> m;
	while (m--) {
		int x, y;
		char tp;
		cin >> tp >> x >> y;
		if (tp == 'F') merge(x, y);
		else {
			e[x].push_back(y);
			e[y].push_back(x);
		}
	}
	for (int i = 1; i <= n; i++) {
		for (vector<int>::iterator it = e[i].begin(); it != e[i].end(); it++) {
			for (vector<int>::iterator it2 = e[*it].begin(); it2 != e[*it].end(); it2++) {
				merge(i, *it2);
			}
		}
	}
	for (int i = 1; i <= n; i++) {
		flag[getfa(i)] = 1;
	}
	int cnt = 0;
	for (int i = 1; i <= n; i++) {
		if (flag[i] == 1) cnt++;
	}
	cout << cnt << endl;
}