poj 1733 并查集

View Code

需要离散化，但用map就方便多了了

#include<stdio.h>
#include<string.h>
#include<map>
using namespace std;
const int maxn = 10010;
int rank[maxn],p[maxn];
int n,m;
void init()
{
    int i;
    for(i=0;i<=10000;i++)
    {
        p[i]=i;
        rank[i]=0;
    }
}
int find(int x)
{
    if(p[x]==x)
       return x;
    int tx=find(p[x]);
    rank[x]=(rank[x]+rank[p[x]])%2;
    p[x]=tx;
    return p[x];
}
int unio(int x,int y,int d)
{
    int tx=find(x);
    int ty=find(y);
    if(tx==ty)
    {
        if((rank[x]+rank[y])%2==d)
             return 1;
       else    return 0;
    }
    else 
    {
        p[tx]=ty;
        rank[tx]=(rank[x]+rank[y]+d)%2;
        return 1;
    }
}
int main()
{
    int i,j,k;
    int a,b;
    char str[10];
    scanf("%d%d",&n,&m);
        init();
        map<int,int> mm;
        int tot=0,d;
        int ans=0;
        while(m--)
        {
            scanf("%d%d%s",&a,&b,str);a--;
            if(mm.find(a)==mm.end())
            {
                mm[a]=tot++;
            }
            if(mm.find(b)==mm.end())
            {
                mm[b]=tot++;
            }
            int x=mm[a],y=mm[b];
            if(str[0]=='o')
                   d=1;
            else d=0;
            if(unio(x,y,d))
            ans++;
            else break;     
        }
        printf("%d\n",ans);  
    return 0;
}
/*
10
5
1 2 even
3 4 odd
5 6 even
1 6 even
7 10 odd
3
*/