2597 团伙（并查集）

╰+攻爆jí腚メ 2022-06-15 00:52 111阅读 0赞

题目描述 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

数据范围及提示 Data Size & Hint

2<=N<=1000

1<=M<=5000

1<=p q<=N

也是一个并查集的模板题，多了一个敌人的敌人是朋友，学习一下codevs题解的做法，用一个数组记录敌人是谁

#include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<algorithm>
    using namespace std;
    int pre[5010],n,q,p;
    void init()
    {
    	for (int i=1; i<=n; i++)
    	{
    		pre[i] = -1;
    	}
    }
    int find(int num)
    {
    	while (pre[num]>-1)
    	{
    		num = pre[num];
    	}
    	return num;
    }
    int union_set(int x, int y)
    {
    	int xx = find(x);
    	int yy = find(y);
    	if (xx != yy)
    	{
    		pre[xx] += pre[yy];
    		pre[yy] = xx;
    	}
    }
    int main()
    {
    	int m,flag[5010]={0},v[5010]={0},cnt=0;
    	char ch;
    	cin>>n>>m;
    	init();
    	for (int i=0 ;i<m; i++)
    	{
    		cin>>ch>>q>>p;
    		if(ch == 'F') //两个人是朋友直接并  
    		{
    			union_set(q,p);
    		}
    		else //两个人是敌人，用标记数据记录，将敌人的敌人并入 
    		{
    			if (flag[q]) //如果q是有敌人的， 那么p就要和q的敌人结为朋友 
    			{
    				union_set(p,flag[q]);
    			}
    			if (flag[p]) //如果p是有敌人的，那么q就要和p的敌人成为朋友
    			{
    				union_set(q,flag[p]);
    			}
    			flag[q] = p;
    			flag[p] = q;
    		}
    	}
    	for(int i=1; i<=n; i++)
    	{
    		if(!v[find(i)])
    		{
    			v[find(i)] = 1;
    			cnt++;
    		}
    	}
    	cout<<cnt;
        return 0;
    }