带权并查集

带权并查集

为什么要用带权并查集
普通并查集只能解决集合间的关系，判断元素属不属于一个集合和一共有多少个集合
带权并查集可以判断集合内部元素之间的关系，甚至是元素之间的权值

题目

1.食物链

题目链接：http://poj.org/problem?id=1182.
动物王国中有三类动物A,B,C，这三类动物的食物链构成了有趣的环形。A吃B， B吃C，C吃A。
现有N个动物，以1－N编号。每个动物都是A,B,C中的一种，但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述：
第一种说法是"1 X Y"，表示X和Y是同类。
第二种说法是"2 X Y"，表示X吃Y。
此人对N个动物，用上述两种说法，一句接一句地说出K句话，这K句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。
1） 当前的话与前面的某些真的话冲突，就是假话；
2） 当前的话中X或Y比N大，就是假话；
3） 当前的话表示X吃X，就是假话。
你的任务是根据给定的N（1 <= N <= 50,000）和K句话（0 <= K <= 100,000），输出假话的总数。
Input
第一行是两个整数N和K，以一个空格分隔。
以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。
若D=1，则表示X和Y是同类。
若D=2，则表示X吃Y。
Output
只有一个整数，表示假话的数目。

题目分析

3类动物，A吃B，B吃C,C吃A构成环
1.数组定义
Fa[]数组表示集合之间的关系，初始化为它本身，表示最开始自己是自己的根节点
cnt[]数组表示集合内元素之间的关系，全部初始化为0，表示自己和自己是同类。i节点到根节点的权值：cnt[i]=0–同类、cnt[i]=1–吃、cnt[i]=2–被吃
2.路径压缩
cnt[]记录的是每个节点到根节点的权值，所以压缩路径时cnt[]要更新

2的根节点是1，3的根节点是2，现要将3的根节点更新为1
如图：cnt[3] (红)=(cnt[2]+cnt[3])%3;
模3可以将cnt[]控制在0，1，2中，便于观察三类动物的关系。若无法理解公式，可以举出几个关系套进公式进行理解

int findfather(int x)
{
    if(fa[x]!=x)
	{
	    int temp=fa[x];
	    fa[x]=findfather(fa[x]);
	    cnt[x]=(cnt[temp]+cnt[x])%3;
	}
	return fa[x];
}

3.合并

根节点不同，如左图
由向量加减可得cnt[ty]=(cnt[x]+m-cnt[y]+3)%3;
根节点相同，如右图，则y->x有
cnt[y->x]=(cnt[y]-cnt[x]+3)%3;
注：+3时防止前面减法出现负数

int unionn(int x,int y,int m)//m=m-1
{
	int tx=findfather(x);
	int ty=findfather(y);
	if(tx==ty)
	{
		if(m!=(cnt[y]-cnt[x]+3)%3)
			return 1;
	}
        fa[ty]=tx;
        cnt[ty]=(cnt[x]+m-cnt[y]+3)%3;
        return 0;

}

4.总体代码

#include <iostream>
#include<stdio.h>
#include<cmath>
#include<algorithm>
#define me(a,b) memset(a,b,sizeof(a))
const int maxn =1e5+10;
using namespace std;
int n,k,fa[maxn],cnt[maxn];
int findfather(int x)
{
    if(fa[x]!=x)
	{
	    int temp=fa[x];
	    fa[x]=findfather(fa[x]);
	    cnt[x]=(cnt[temp]+cnt[x])%3;
	}
	return fa[x];
}
int unionn(int x,int y,int m)
{
	int tx=findfather(x);
	int ty=findfather(y);
	if(tx==ty)
	{
		if(m!=(3+cnt[y]-cnt[x])%3)//x与y已有的关系与现在所给的关系m不相符
			return 1;
		else	return 0;
	}
        fa[ty]=tx;
        cnt[ty]=(cnt[x]+m+3-cnt[y])%3;
        return 0;
}
int main()
{
	scanf("%d%d",&n,&k);
		for(int i=0;i<=n;i++)//赋初值
		{
			fa[i]=i;
			cnt[i]=0;
		}
		int ans=0;//赋初值，记录假话个数
		while(k--)
		{
			int m,x,y;
			scanf("%d%d%d",&m,&x,&y);
            if(x>n||y>n)	ans++;//假话条件2
			else if(m==2&&x==y)	ans++;//假话条件3
			else//假话条件1
			{
				ans+=unionn(x,y,m-1);
			}
		}
		printf("%d\n",ans);
    return 0;
}