Luogu P2024&P1892食物链&团伙（并查集拓展域）

Posted 2020-11-21 notscience

Luogu P1892
Luogu P2024
这两道一眼看过去很容易发现可以用并查集来做——但是当我们仔细阅读题面后，会发现其实并没有那么简单。
我们知道并查集可以很轻松地维护具有传递性的信息，也就是“朋友的朋友就是我的朋友”这样的关系，但是普通的并查集并不能维护“敌人的敌人是朋友”这种关系。
这时候我们就要引入一种神奇的操作，将并查集扩大一倍，将增加的这一倍空间来维护节点i的敌人。
例如对于团伙这一题

    if (c=='F')
        {
            merge(x,y);
        }
        else 
        {
            merge(x,y+n);
            merge(y,x+n);
            //x+n代表的是x的敌人集合
        }

非常好理解，确实是一种相当高效的操作。

P1892完整代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=2001;
int fa[maxn],n,m,cnt,x,y;
int getf(int v)
{
    if (fa[v]==v) return fa[v];
    return fa[v]=getf(fa[v]);
}
inline void merge(int x,int y)
{
    fa[getf(y)]=getf(x);
}
inline bool check(int x,int y)
{
    if (getf(x)==getf(y)) return true;
    return false;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=2*n;i++) fa[i]=i;
    for (int i=1;i<=m;i++)
    {
        char c;
        scanf("
%c %d %d",&c,&x,&y);
        if (c=='F')
        {
            merge(x,y);
            //merge(x+n,y+n);
            //注意上面这句不能要，因为题目中没有说到朋友的敌人也是我的敌人。
        }
        else 
        {
            merge(x,y+n);
            merge(y,x+n);
        }
    }
    for (int i=1;i<=n;i++) fa[i]=getf(i);//路径压缩，防止压缩不完全
    for (int i=1;i<=n;i++) if (fa[i]==i) cnt++;
    printf("%d
",cnt);
    return 0;
}

P2024完整代码

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=3*5*1e4+10;
int fa[maxn],cnt,n,k,flag,x,y;
int getf(int v)
{
    if (fa[v]==v) return v;
    return fa[v]=getf(fa[v]);
}
inline void merge(int x,int y)
{
    x=getf(x);
    y=getf(y);
    fa[x]=y;
}
inline bool check(int x,int y)
{
    x=getf(x);
    y=getf(y);
    if (fa[x]==fa[y]) return true;
    return false;
}
int main()
{
    scanf("%d%d",&n,&k);
    for (int i=1;i<=3*n;i++)
        fa[i]=i;
    //i+n是i吃的集合，i+2n是吃i的集合
    for (int i=1;i<=k;i++)
    {
        scanf("%d%d%d",&flag,&x,&y);
        if (x>n||y>n) {cnt++;continue;}
        if (flag==1)
        {
            if (check(x+n,y)) {cnt++;continue;}
            if (check(x+2*n,y)) {cnt++;continue;}
            merge(x,y);
            merge(x+n,y+n);
            merge(x+2*n,y+2*n);
        }
        if (flag==2)
        {
            if (check(x,y)) {cnt++;continue;}
            if (check(y+n,x)) {cnt++;continue;}
            merge(x+n,y);
            merge(y+2*n,x);
            merge(x+2*n,y+n);
        }
    }
    printf("%d
",cnt);
    return 0;
}