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;
}