POJ 1182 食物链（并查集）

最新推荐文章于 2022-12-22 21:28:01 发布

原创最新推荐文章于 2022-12-22 21:28:01 发布 · 330 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#poj

并查集专栏收录该内容

0 篇文章

订阅专栏

本文介绍了一个使用并查集算法解决复杂数据校验问题的例子。通过处理三种不同类型的元素及其相互之间的关系，确保数据的一致性和正确性。文章详细解释了如何初始化并查集、查找根节点以及合并操作，并给出了完整的C++实现代码。

三倍的par数组，每一个部分分别表示A，B，C；
对于每一次数据判断是否正确，总共3*3中情况，六种
错误，剩下三组对的就并在一起，输入数据的时候，
千万不要用while输入多组，一组就够了，20发Wa
后的领悟。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;

int n,k;
#define maxn 150001
int par[maxn],rank[maxn],T[maxn];
int X[maxn],Y[maxn];

void init(int p)
{
    for(int i=0;i<=p;i++)
    {
        par[i]=i;
        rank[i]=0;
    }
}

int find(int x)
{
    if(par[x]==x) 
        return x;
    else 
        return par[x]=find(par[x]);
}

void unite(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y) return;
    if(rank[x]<rank[y])
        par[x]=y;
    else
    {
        par[y]=x;
        if(rank[x]==rank[y])
            rank[x]++;
    }
}

bool same(int x,int y)
{
    return find(x) == find(y);
}

void solve()
{
    init(n*3);
    int ans=0;
    for(int i=0;i<k;i++)
    {
        int t=T[i];
        int x=X[i]-1, y=Y[i]-1;
        if(x<0||n<=x||y<0||n<=y)
        {
            ans++;
            continue;
        }
        if(t==1)
        {
            if(same(x,y+n)||same(x,y+2*n)||same(x+n,y)||same(x+2*n,y)
            ||same(x+n,y+2*n)||same(x+2*n,y+n))
                ans++;
            else
            {
                unite(x,y);
                unite(x+n,y+n);
                unite(x+2*n,y+2*n);
            }
        }
        else
        {
            if(same(x,y)||same(x,y+2*n)||same(x+n,y+n)||same(x+2*n,y+2*n)
            ||same(x+n,y)||same(x+2*n,y+n))
                ans++;
            else
            {
                unite(x,y+n);
                unite(x+n,y+2*n);
                unite(x+2*n,y);
            }
        }
    }
    printf("%d\n",ans);
}

int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<k;i++)
    {
        scanf("%d%d%d",T+i,X+i,Y+i);
    }
    solve();
    return 0;
}