poj 1182 并查集

第一次wa是因为负数的问题

现在虽然ac了 不过还是有一点小问题

赶着睡觉

晚上再处理

 

 

#include <iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
const int maxn=50001;
struct
{
    int fa,re;
}f[maxn];
int ans;

int findre(int x)
{
    if(f[x].fa!=x)
    return((f[x].re+findre(f[x].fa)+3)%3);
    return((f[x].re+3)%3);
}

int find(int x)
{
    if(f[x].fa!=x)
    {
        f[x].re+=findre(f[x].fa);
        f[x].re%=3;
        f[x].fa=find(f[x].fa);
    }
    return(f[x].fa);
}

int main()
{
//    freopen("in.txt","r",stdin);
    int n,k;
    scanf("%d %d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        f[i].fa=i;
        f[i].re=0;
    }
    ans=0;
    for(int i=1,a,b,c;i<=k;i++)
    {
        scanf("%d %d %d",&c,&a,&b);
        if(a>n||b>n)
        {
            ans++;
            continue;
        }
        else if(c==1)
        {
            int f1=find(a);
            int f2=find(b);
            if(f1!=f2)
            {
                f[f2].re=findre(a)-findre(b);
                f[f2].fa=f1;
            }
            else
            {
                int r1=findre(a);
                int r2=findre(b);
                if(r1%3!=r2)
                ans++;
            }
        }
        else
        {
            int f1=find(a);
            int f2=find(b);
            if(f1!=f2)
            {
                f[f2].re=findre(a)+1-findre(b);
//                if(f[f2].re<0)
//                f[f2].re+=-(f[f2].re/3*3)+9;
                f[f2].fa=f1;
            }
            else
            {
                int r1=findre(a);
                int r2=findre(b);
                if((r1+1)%3!=r2)
                ans++;
            }
        }
    }

//    for(int i=1;i<=n;i++)
//    printf("%d %d\n",find(i),findre(i));

    printf("%d\n",ans);
    return 0;
}