LDUOJ 团伙#并查集

团伙

思路：
这是一个典型的并查集的题目
按照题意，如果x和y是朋友，自然属于一个集合
如果x和y是敌人，那么x的所有敌人都是y的朋友
这时就需要合并x的敌人和y了
我用一个二维数组来记录x和y的敌人关系 fight[x][y]
这样 fight[x][i]就是x的敌人，也就是y的朋友
最后查找一下根节点就可以了

int n,m,p,x,y;
int ans;
int f[1010];//记录团伙头目 
bool fight[1010][1010];//记录敌人关系 
inline int find(int x)//寻找根节点 
{
    if (f[x] != x) x = f[x] = find(f[x]);//路径压缩
    return x;
}

void merge(int x,int y)//合并x和y所在集合
{
    int f1 = find(x);
    int f2 = find(y);
    if (f1 != f2)	f[f2] = f1;
}

int main()
{
    cin >> n >> m;
    for(int i=1;i<=n;i++)	f[i] = i; //初始化并查集 
    for(int k=1;k<=m;k++) 
    {
        cin >> p >> x >> y;
        if(p == 0)	merge(x,y); //如果是朋友直接合并
        else //否则是敌人  
        {
            fight[x][y] = fight[y][x] = true; //敌人关系
            for(int i=1;i<=n;i++)
            {
                if(fight[x][i])	merge(y,i); 
				//x和i是敌人，又因为x和y是敌人，所以y和i是朋友 
                if(fight[y][i])	merge(x,i); //同上
            }
        }
    }
    for(int i=1;i<=n;i++)
        if(f[i] == i)
            ans++; //统计团伙个数（也就是树根）
    cout << ans << endl;
    return 0;
}