poj 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
只有一个整数，表示假话的数目。
Sample Input
100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5
#include <stdio.h>


#define MAX 50001


int count;        //记录谎话的数目


int N,K;            //整数N,K表示动物个数以及说话次数
int D,X,Y;        //D代表动物X，Y之间的关系


struct
{
    int x;
    int y;


}a[MAX];


/*********  初始化数组  *********/
void Initial()
{
    int i;


    for(i=1;i<MAX;i++)
    {
        a[i].x=i;
        a[i].y=0;
    }
}


/*********  查找根节点函数  *********/
int find(int x)
{
    if(a[x].x==x)
    {
        return x;
    }


    a[x].x=find(a[x].x);        //压缩路径


    a[x].y=(a[x].y+a[a[x].x].y)%3;        //重新定义子节点与根节点的关系
    
    return a[x].x;
}


/**********  合并两个集合  *********/
void Union(int x,int y,int rx,int ry)
{
    a[ry].x=rx;
    a[ry].y=(3-a[y].y+D-1+a[x].y)%3;        //当合并了集合后更新节点的关系
}


/*********  判断话是否谎话  *********/
void Judge(int x,int y)
{
    int rx,ry;


    rx=find(x);
    ry=find(y);


    if(x>N || x<1 || y>N || y<1 || (x==y && D==2))
    {
        count++;
        return;
    }


    Union(x,y,rx,ry);
    
    switch(D)
    {
    case 1:
        if(a[x].y!=a[y].y)
        {
            count++;
        }
        break;
    case 2:
        if(((a[y].y+(3-a[x].y))%3)!=1)
        {
            count++;
        }
    break;
    }
}


int main()
{
    int a;


    int i;
    
    while(scanf("%d",&a)!=EOF)
    {
        Initial();
        
        count=0;


        for(i=0;i<a;i++)
        {
            scanf("%d%d",&N,&K);


            while(K!=0)
            {
                scanf("%d%d%d",&D,&X,&Y);


                Judge(X,Y);


                K--;
            }
            
            printf("%d\n",count);
        }
    }
    return 0;
}