HDU 1811 Rank of Tetris 并查集+拓扑排序

我这号菜鸡，上来又用一个蠢萌的算法WA掉了这道题，就是把相同级别的点，也按照编号的大小做一个拓扑排序。其实这用手指头肚想想都不可能对，一定会把原本conflict的情况算成ok的。所以转而使用正常人会用的方法，使用并查集把相等级别的点合成一个点，这样很好。

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<string>
#include<queue>
#include<vector>
using namespace std;

int fa[10005],n,m,d[10005],sum,x[20010],y[20010];
bool vis[10005];
vector<int> vec[10005];
char o[200010];

int find(int x)
{
    int fx=fa[x];
    if (fa[x]!=x)
    {
        fx=find(fa[x]);
    }
    return fa[x]=fx;
}

bool u(int a,int b)
{
    int fx=find(a),fy=find(b);
    if (fx==fy) return false;
    fa[fy]=fx;
    return true;
}

int main()
{
    while (~scanf("%d%d",&n,&m))
    {
        memset(d,0,sizeof(d));
        sum=n;
        for (int i=0; i<=n; i++)
        {
            vec[i].clear(); fa[i]=i;
        }
        for (int i=1; i<=m; i++)
        {
            scanf("%d %c %d",&x[i],&o[i],&y[i]);
            if (o[i]=='=')
            {
                if (u(x[i],y[i])) sum--;
            }
        }
        for (int i=1; i<=m; i++)
        {
            if (o[i]=='=') continue;
            int X=find(x[i]);
            int Y=find(y[i]);
            if (o[i]=='>')
            {
                vec[X].push_back(Y);
                d[Y]++;
            }else
            {
                vec[Y].push_back(X);
                d[X]++;
            }
        }
        int flag=0;
        queue<int> stc;
        for (int i=0; i<n; i++)
        {
            if (find(i)==i && d[i]==0)
            {
                    stc.push(i);
            }
        }
        while (!stc.empty())
        {
            if (stc.size()>1) flag=1;
            int nod=stc.front();
            stc.pop();
            sum--;
            //cout<<"nod="<<nod<<endl;
            for (int i=0; i<vec[nod].size(); i++)
            {
                int v=vec[nod][i];
                d[v]--;
                if (d[v]==0) stc.push(v);
                //cout<<"v="<<v<<" dv="<<d[v]<<endl;
            }
        }
        if (sum>0) cout<<"CONFLICT"<<endl;
        else if (flag==0) cout<<"OK"<<endl;
        else if (flag==1) cout<<"UNCERTAIN"<<endl;
    }
    return 0;
}