HDU1811 并查集+拓扑排序

题目大意：

判断是否能根据给定的规则将这一串数字准确排序出来

我们用小的数指向大的数

对于相等的情况下，将二者合并到同一个并查集中，最后抽象出来的图上面的每一个点都应该代表并查集的标号

#include <cstdio>
#include <cstring>
#include <stack>
#include <iostream>
using namespace std;

#define N 10005
char s[N<<1][3];
int fa[N] , first[N] , in[N] , k , cnt;//fa[i]保存的是i点处于的并查集号,cnt记录最后剩下的并查集数
bool vis[N];//判断对应下标的集合是否会进入最后生成的图中
struct Edge{
    int y,next;
}e[N<<1];

void add_edge(int x,int y)
{
    in[y]++;
    e[k].y = y , e[k].next=first[x];
    first[x] = k++;
}

void init(int n)
{

    k=0;
    memset(first , -1 , sizeof(first));
    memset(in,0,sizeof(in));
    for(int i=0;i<n;i++)
        fa[i] = i , vis[i] = true;
}

int find_fa(int x)
{
    while(x!=fa[x])
        x=fa[x];
    return x;
}

void Union(int a , int b)
{
    int fa1_set = find_fa(a);
    int fa2_set = find_fa(b);
    if(fa1_set != fa2_set)
    {
        fa[fa2_set] = fa1_set , cnt--;
        vis[fa2_set] = false; //fa2_set集合被合并，所以最后不会进入图中，所以将其舍去
    }
}

int tuopu(int n)
{
    stack<int> s;
    for(int i = 0 ; i<n ; i++){
        if(vis[i] && !in[i]) s.push(i);
    }
    int flag = 1;
    for(int i=0 ; i<cnt ; i++){
        if(s.empty()) return -1;
        if(s.size() > 1) flag = 0;
        int u = s.top();
        s.pop();
        for(int j=first[u] ; j!=-1 ; j=e[j].next){
            int v = e[j].y;
            in[v]--;
            if(!in[v]){
                s.push(v);
            }
        }
    }
    if(!flag) return 0;
    return 1;
}

int main()
{
   // freopen("a.in" , "rb" ,stdin);
    int n , m , a[N] , b[N];
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init(n);
        cnt = n;

        for(int i=0;i<m;i++){
            scanf("%d%s%d",&a[i],s[i],&b[i]);
            if(s[i][0] == '='){
                Union(a[i],b[i]);
            }
        }
        //cout<<"cnt: "<<cnt<<endl;
        for(int i = 0 ; i<m ; i++){
            int fa1 = find_fa(a[i]);
            int fa2 = find_fa(b[i]);
            if(s[i][0] == '<'){
                add_edge(fa1 , fa2);
            }
            else if(s[i][0] == '>'){
                add_edge(fa2 , fa1);
            }
        }
        int flag = tuopu(n);
        if(flag == 1) puts("OK");
        if(flag == -1) puts("CONFLICT");
        if(flag == 0) puts("UNCERTAIN");
    }
    return 0;
}