洛谷 P1892 [BOI2003]团伙 并查集变形

题目链接：

https://www.luogu.com.cn/problem/P1892

参考博客：

https://www.luogu.com.cn/blog/user52892/solution-p1892

自己的思路几乎完全是参照上边这个博客的思路来的，上边的博客讲解的很详细，可以再次参考

算法：

并查集

核心代码：

    for(int i=1;i<=m;i++)
    {
        cin>>ch>>a>>b;
        if(ch=='F')union_(a,b);
        else
        {
            fight[a][b]=fight[b][a]=true;
            for(int i=1;i<=n;i++)
            {
                if(fight[a][i])union_(b,i);
                if(fight[b][i])union_(a,i);
            }
        }
    }

1：fight[maxn][maxn]存仇敌关系

2：如果是朋友直接合并

        if(ch=='F')union_(a,b);

3：如果是敌人，先记录仇敌关系

            fight[a][b]=fight[b][a]=true;

4：敌人的敌人是朋友

            for(int i=1;i<=n;i++)
            {
                if(fight[a][i])union_(b,i);
                if(fight[b][i])union_(a,i);
            }

5：树根的个数也就是团伙个数

#include <bits/stdc++.h>

using namespace std;
const int maxn=1e3+1;
int fa[maxn],n,m,a,b;
char ch;
bool fight[maxn][maxn];

void init()
{
    for(int i=1;i<=n;i++)fa[i]=i;
}

int find(int x)
{
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}

void union_(int x,int y)
{
    int f1=find(x);
    int f2=find(y);
    if(f1!=f2) fa[f2]=f1;
}

int main()
{
    ios::sync_with_stdio(0);
    cin>>n>>m;
    init();
    for(int i=1;i<=m;i++)
    {
        cin>>ch>>a>>b;
        if(ch=='F')union_(a,b);
        else
        {
            fight[a][b]=fight[b][a]=true;
            for(int i=1;i<=n;i++)
            {
                if(fight[a][i])union_(b,i);
                if(fight[b][i])union_(a,i);
            }
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
        if(fa[i]==i)ans++;
    cout<<ans<<endl;
    return 0;
}