HDU 1181

拓扑排序+并查集

/*#include<bits/stdc++.h>
using namespace std;*/
#include <stdio.h>
#include <string.h>
#include<iostream>
#include <vector>
#include <queue>
using namespace std;
#define maxn 10005
int pre[10005];
vector<int> G[maxn];
int in[maxn];           //入度
int topo[maxn], t, n,m;
int sum;
bool toposort()
{
    priority_queue<int, vector<int>, greater<int> > q;
   // queue<int> q;
    t = 0; //优先队列保证最小序列
    for(int i = 0; i <n; ++i)
    {
        if(in[i] == 0&&i==pre[i]) q.push(i);
    }
    int flag=0;
    int cnt = 0;
    while(!q.empty())
    {
        if(q.size()>1)
        {
            flag=1;
        }
        int u = q.top();
        q.pop();
        sum--;
        topo[t++] = u;
        ++cnt;
        for(int v = 0; v < G[u].size(); ++v)
        {
            if(--in[G[u][v]] == 0) q.push(G[u][v]);             //找入度为0的边入队
        }
    }
    if(sum>0)
    {
        cout<<"CONFLICT"<<endl;
    }
    else if(flag)
    {
        cout<<"UNCERTAIN"<<endl;
    }
    else
        cout<<"OK"<<endl;
}
int find(int x)
{
    int p,t;
    p=x;
    while(x!=pre[x])
    {
        x=pre[x];
    }
    while(p!=x)
    {
        t=pre[p];
        pre[p]=x;
        p=t;
    }
    return x;
}
bool join(int a,int b)
{
    a=find(a);
    b=find(b);
    if(a!=b)
    {
        pre[b]=a;
        return true;
    }
    return false;
}
struct ci
{
    int a,b;
    char c;
}no[maxn];
int main()
{
    while(cin>>n>>m)
    {
        for(int i=0;i<maxn;i++)
        {
            pre[i]=i;
        }
        for(int i=0;i<maxn;i++)
        {
            G[i].clear();
        }

        memset(in,0,sizeof(in));
        memset(topo,0,sizeof(topo));
        sum=n;
        for(int i=0; i<m; i++)
        {
            cin>>no[i].a>>no[i].c>>no[i].b;
            if(no[i].c=='=')
            {
                if(join(no[i].a,no[i].b))
                {
                    sum--;
                }
            }
        }
        for(int i=0;i<m;i++)//1️⃣此for循环必须写到外面
        {
            if(no[i].c=='=')
                continue;
            int p=find(no[i].a);//pre[no[i].a];//2️⃣这里是find()而不能pre。此两点导致我WA无数次
            int q=find(no[i].b);//pre[no[i].b];
            if(no[i].c=='>')
            {
                G[p].push_back(q);
                in[q]++;
            }
            else
            {
                G[q].push_back(p);
                in[p]++;
            }
        }
        toposort();
    }
    return 0;

}