poj——1182食物链 并查集（提升版）

因为是中文题，题意就不说了，直接说思路：

我们不知道给的说法中的动物属于A B C哪一类，所以我们可以用不同区间的数字表示这几类动物,这并不影响结果，我们可以用并查集把属于一类的动物放在一块，举个例子：

A    2*n ~ 3*n-1

B    n ~ 2*n-1

C   0 ~ n-1

假设 n =10，1和3是同类，那么1和3并到一块，11和13并到一块，21和23并到一块。那么这样既不相互矛盾，又把这三种动物的可能性列举了出来。1捕食4，那么说明1比4高了一个等级，将1和14归并为一类，11和24归并为一类，21和4归并为一类，即B吃C,A吃B,C吃A。

看代码：

#include<stdio.h>
int f[160001];
int getf(int v)
{
    if(f[v]==v)
        return v;
    return f[v]=getf(f[v]);
}
void marge(int u,int v)
{
    int t1=getf(u),t2=getf(v);
    if(t1!=t2)
        f[t2]=t1;
}
bool same(int u,int v)
{
    int t1=getf(u),t2=getf(v);
    if(t1!=t2)
        return 0;
    return 1;
}
int main()
{
//A 2*n ~ 3*n-1
//B n ~ 2*n-1
//C 0~ n-1
    int n,m,sum,a,b,p;
    scanf("%d%d",&n,&m);
    for(int i=0; i<n*3+10; i++)
        f[i]=i;
    sum=0;
    for(int i=0; i<m; i++)
    {
        scanf("%d%d%d",&p,&a,&b);
        a-=1,b-=1;
        if(a>=n||b>=n||b<0||a<0)
        {
            sum++;
            continue;
        }
        if(p==1)
        {
            if(same(a,b+n)||same(a,b+2*n))//a和b不能有捕食和被捕食的关系
                sum++;
            else
            {
                marge(a,b);//同一等级
                marge(a+n,b+n);
                marge(a+2*n,b+2*n);
            }
        }
        else if(p==2)
        {
            if(same(a,b)||same(a,b+2*n))//a和不能是同类 也不能是b捕食a
                sum++;
            else//a捕食b表示a和b+n是同一等级
            {
                marge(a,b+n);
                marge(a+n,b+2*n);
                marge(a+2*n,b);
            }
        }
    }
    printf("%d\n",sum);
    return 0;
}