1. 程式人生 > >【並查集】團夥

【並查集】團夥

() des stdin scrip blog -a main amp 產生

[codevs2597] 團夥

時間限制: 1 s 空間限制: 128000 KB 題目等級 : 黃金 Gold
題目描述 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

試題分析:這種問題我們一般有兩種解法:①打標記 ②多個並查集

     打標記這類方法會在銀河英雄傳說中看到

     那麽多個並查集如何解決呢?我們設1~N節點是記錄朋友,N+1~2*N節點是記錄敵人

     既然兩個人是朋友,根據題意,我們也要把他們的敵人合起來

     如果兩個人是敵人,那麽merge(a,b+N),merge(a+N,b)

     如何查詢有幾個團夥呢,就把所有人的根節點放進一個數組裏面sort一下最後求下多少不同的就好了

代碼:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<stack>
#include<vector>
#include<algorithm>
//#include<cmath>

using namespace std;
const int INF = 9999999;
#define LL long long

inline int read(){
	int x=0,f=1;char c=getchar();
	for(;!isdigit(c);c=getchar()) if(c==‘-‘) f=-1;
	for(;isdigit(c);c=getchar()) x=x*10+c-‘0‘;
	return x*f;
}
int fa[3001]; 
int a[3001];
int N,M;
int ans;

void init(){
	for(int i=1;i<=2*N;i++) fa[i]=i;
}
int find(int x){
    if(x!=fa[x]) return fa[x]=find(fa[x]);
    return x;
}
void merge(int a,int b){
	int x=find(a),y=find(b);
	if(x==y) return ;
	fa[y]=x;
	return ;
}
int main(){
	//freopen(".in","r",stdin);
	//freopen(".out","w",stdout);
	N=read(),M=read();
	init(); 
	for(int i=1;i<=M;i++){
		char c;cin>>c;
		int a=read(),b=read();
		if(c==‘F‘){
			merge(a,b);
		}
		else{
			merge(a+N,b);
			merge(a,b+N);
		}
	}
	for(int i=1;i<=N;i++) a[i]=find(i);
	sort(a+1,a+N+1);
	for(int i=1;i<=N;i++) if(a[i]!=a[i-1]) ans++;
	printf("%d\n",ans);
	return 0;
}

【並查集】團夥