[BZOJ1370][Baltic2003]Gang团伙

[BZOJ1370][Baltic2003]Gang团伙

试题描述

在某城市里住着n个人，任何两个认识的人不是朋友就是敌人，而且满足： 1、 我朋友的朋友是我的朋友； 2、 我敌人的敌人是我的朋友； 所有是朋友的人组成一个团伙。告诉你关于这n个人的m条信息，即某两个人是朋友，或者某两个人是敌人，请你编写一个程序，计算出这个城市最多可能有多少个团伙？

输入

第1行为n和m，N小于1000,M小于5000； 以下m行，每行为p x y，p的值为0或1，p为0时，表示x和y是朋友，p为1时，表示x和y是敌人。

输出

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

输入示例

E
F
F
E  

输出示例

数据规模及约定

见“输入”

题解

朋友关系直接加入并查集；敌对关系建一张无向图，扫一遍每个点，在无向图上与它直接相连的点依次在并查集中合并。最后输出连通分量个数。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <stack>
#include <vector>
#include <queue>
#include <cstring>
#include <string>
#include <map>
#include <set>
using namespace std;

const int BufferSize = 1 << 16;
char buffer[BufferSize], *Head, *Tail;
inline char Getchar() {
	if(Head == Tail) {
		int l = fread(buffer, 1, BufferSize, stdin);
		Tail = (Head = buffer) + l;
	}
	return *Head++;
}
int read() {
	int x = 0, f = 1; char c = Getchar();
	while(!isdigit(c)){ if(c == '-') f = -1; c = Getchar(); }
	while(isdigit(c)){ x = x * 10 + c - '0'; c = Getchar(); }
	return x * f;
}

#define maxn 1010
#define maxm 10010
int n, m, M, head[maxn], next[maxm], to[maxm], pos[maxn];

void AddEdge(int a, int b) {
	to[++M] = b; next[M] = head[a]; head[a] = M;
	swap(a, b);
	to[++M] = b; next[M] = head[a]; head[a] = M;
	return ;
}

int fa[maxn];
int findset(int x) { return x == fa[x] ? x : fa[x] = findset(fa[x]); }

int main() {
	n = read(); m = read();

	for(int i = 1; i <= n; i++) fa[i] = i;
	while(m--) {
		char tc = Getchar();
		while(!isalpha(tc)) tc = Getchar();
		if(tc == 'E') {
			int a = read(), b = read();
			AddEdge(a, b);
		}
		if(tc == 'F') {
			int u = findset(read()), v = findset(read());
			if(u != v) fa[v] = u;
		}
	}
	for(int i = 1; i <= n; i++) {
		int le = head[i], e;
		for(e = next[le]; e; e = next[e]) {
			int u = findset(to[le]), v = findset(to[e]);
			if(u != v) fa[v] = u;
			le = e;
		}
	}

	for(int i = 1; i <= n; i++) pos[i] = findset(i);
	sort(pos + 1, pos + n + 1);
	int cnt = 0;
	for(int i = 1; i <= n; i++) if(i == 1 || pos[i] != pos[i-1]) cnt++;
	printf("%d\n", cnt);

	return 0;
}