HDU1811 Rank of Tetris （并查集）

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string> 
#include<set>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int in[N];
int x[N],y[N];
int p[N];
string s[N];
int n,m;
int h[N],e[N],ne[N],idx;
int sum;
void add(int a,int b){ //前向星 
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void init(){
    memset(h,-1,sizeof h);
    idx=0;
    memset(in,0,sizeof in);
}
int find(int x){
    if(p[x]!=x){
        p[x]=find(p[x]);
    }
    return p[x];
}
void topo(){
    int i;
    queue<int> q;
    for(i=0;i<n;i++){
        if(!in[i]&&p[i]==i)//一定要找根节点且入度为0，不然多加了点 
        q.push(i);
    }
    if(q.size()==0){
        cout<<"CONFLICT"<<endl;
        return  ;
    }
    int top=0;
    while(q.size()){
        if(q.size()>1){
            top=1; //不能直接break，因为如果冲突加不完全则输出冲突 
        }
        int u=q.front();
        q.pop();
        sum--;
        for(i=h[u];i!=-1;i=ne[i]){
            int j=e[i];
            in[j]--;
            if(!in[j])
            q.push(j);
        }
    }
    if(sum>0)
    cout<<"CONFLICT"<<endl;
    else{
        if(top==1)
        cout<<"UNCERTAIN"<<endl;
        else
        cout<<"OK"<<endl;
    }
}
int main(){
    int i;
    while(cin>>n>>m){
        for(i=0;i<n;i++)
        p[i]=i;
        init();
        sum=n;
        for(i=1;i<=m;i++){
            cin>>x[i]>>s[i]>>y[i];
            if(s[i]=="="){
                int pa=find(x[i]),pb=find(y[i]);
                if(pa!=pb){
                    p[pa]=pb;
                    sum--;//记得缩完点后个数变少 
                }
            }
        }
        int sign=0;
        for(i=1;i<=m;i++){
            int pa=find(x[i]),pb=find(y[i]);
            if(pa==pb&&s[i]!="="){//这个地方说明两个相等的点存在相等又存在不等关系，则矛盾，例如a=b,b=c,a>c,这在之前是检查不出来的 
                sign=1;
                break;
            }
            if(s[i]==">"){
                add(pa,pb);
                in[pb]++;
            }
            else if(s[i]=="<"){
                add(pb,pa);
                in[pa]++;
            }
        }
        if(sign)
        cout<<"CONFLICT"<<endl;
        else
        topo();
    }
}