Rank of Tetris (并查集+拓普排序)

http://acm.hdu.edu.cn/showproblem.php?pid=1811

#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=10000+10;
int n,m;
int p[N],in[N];
int arr[N],brr[N];
char op[N];
int finds(int x) {return x==p[x]? x:p[x]=finds(p[x]);}
int main()
{
    int i,a,b,fa,fb,tn;
    while(cin>>n>>m)
    {
        vector<int> v[N];tn=0;
        queue<int> q;
        char s[5];
        memset(in,0,sizeof(in));
        for(i=0;i<n;i++) p[i]=i;
        for(i=0;i<m;i++)
        {

            scanf("%d%s%d",&a,s,&b);
            arr[i]=a;brr[i]=b;op[i]=s[0];
            fa=finds(a);fb=finds(b);
            if(s[0]=='=')//首先把相等的看成一个集合(一定要先做)
            {
                if(fa!=fb) {p[fa]=fb;tn++;}
            }
        }
        for(i=0;i<m;i++)//建立集合之间的关系(建图)
        {
            fa=finds(arr[i]);
            fb=finds(brr[i]);
            if(op[i]=='<')
            {
                int temp=fa;
                fa=fb;fb=temp;
            }
            if(op[i]!='=')
            {
                if( find( v[fa].begin(), v[fa].end(),fb)==v[fa].end())
                {
                    v[fa].push_back(fb);
                    in[fb]++;
                }
            }
        }
        for(i=0;i<n;i++) if(in[i]==0&&p[i]==i)//所有的树根入队
        {
            q.push(i);
        }
        n-=tn;
        int UNCERTAIN=0;
        while(!q.empty())//拓普排序
        {
            if(q.size()>1) UNCERTAIN=1;//有多个根说明信息不完全
            a=q.front();
            q.pop();
            n--;
            for(i=0;i<v[a].size();i++)
            {
                fa=v[a][i];
                if(--in[fa]==0)
                {
                    q.push(fa);
                }
            }
        }
        if(n>0) printf("CONFLICT\n");//n>0说明有环，冲突
        else if(UNCERTAIN) printf("UNCERTAIN\n");
        else printf("OK\n");
    }
    return 0;
}