poj1182 带权并查集

题意:一共有ABC三种动物,A吃B,B吃C,C吃A,

现在共有n个动物,编号1~n,给出k句话,判断真假;

每句话包含val,u,v;

val==1,代表u,v,是同类,val==2代表u吃v

假的条件为:

1.与前面某些真话冲突;

2.u,v,大于n

3.当val==2时,u!=v, 即不能自己吃自己;

思路:有联系的为一个集合,0代表和集合根同类,1代表吃集合根,2代表被集合根吃;

刚开始是每个动物都是一个集合,根为他们自己;然后进行集合运算,具体看大神博客:https://agatelee.cn/2017/05/带权并查集/

进行集合运算的意义是通过每个动物与根的关系推导每个动物之间的关系,至于规律,看上面链接中的表;

AC代码:

#include 
#include 
#include 
#include 
using namespace std;
const int maxn=1e5+10;
int mes[maxn],fa[maxn],n;
void init(int n)
{
    for (int i=0;i<=n;i++)
    {
        fa[i]=i;
        mes[i]=0;
    }
}
int find(int x)
{
    if (fa[x]==x)
        return x;
    else
    {
        int temp=fa[x];
        fa[x]=find(temp);
        mes[x]=(mes[x]+mes[temp])%3; // 路径压缩
        return fa[x];
    }
}
int check(int u,int v,int val)
{
    if (u>n||v>n)
        return 0;
    if (val==1&&u==v)
        return 0;
    int uu=find(u);
    int vv=find(v);
    if (uu==vv)
        return (mes[u]-mes[v]+3)%3==val; //同根则判断关系是否满足前面的真话所建立的关系
    else
        return 1;
}
void Merge(int u,int v,int val)
{
    int uu=find(u);
    int vv=find(v);
    if (uu!=vv)
    {
        fa[uu]=vv;
        mes[uu]=(mes[v]-mes[u]+val+3)%3; // 不属于同一集合,即根不相同,则改变两根之间的关系;
    }
}
int main()
{
    int k,ans,val,u,v;
    ans=0;
    scanf("%d%d",&n,&k);
    init(n);
    while (k--)
    {
        scanf("%d%d%d",&val,&u,&v);
        val--;
        if (check(u,v,val))
            Merge(u,v,val);
        else
            ans++;
    }
    cout<

 

你可能感兴趣的:(poj1182 带权并查集)