HDU1811 并查集+拓扑排序

题目大意：

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

 

我们用小的数指向大的数

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

 

  1 #include <cstdio>

  2 #include <cstring>

  3 #include <stack>

  4 #include <iostream>

  5 using namespace std;

  6 

  7 #define N 10005

  8 char s[N<<1][3];

  9 int fa[N] , first[N] , in[N] , k , cnt;//fa[i]保存的是i点处于的并查集号,cnt记录最后剩下的并查集数

 10 bool vis[N];//判断对应下标的集合是否会进入最后生成的图中

 11 struct Edge{

 12     int y,next;

 13 }e[N<<1];

 14 

 15 void add_edge(int x,int y)

 16 {

 17     in[y]++;

 18     e[k].y = y , e[k].next=first[x];

 19     first[x] = k++;

 20 }

 21 

 22 void init(int n)

 23 {

 24 

 25     k=0;

 26     memset(first , -1 , sizeof(first));

 27     memset(in,0,sizeof(in));

 28     for(int i=0;i<n;i++)

 29         fa[i] = i , vis[i] = true;

 30 }

 31 

 32 int find_fa(int x)

 33 {

 34     while(x!=fa[x])

 35         x=fa[x];

 36     return x;

 37 }

 38 

 39 void Union(int a , int b)

 40 {

 41     int fa1_set = find_fa(a);

 42     int fa2_set = find_fa(b);

 43     if(fa1_set != fa2_set)

 44     {

 45         fa[fa2_set] = fa1_set , cnt--;

 46         vis[fa2_set] = false; //fa2_set集合被合并，所以最后不会进入图中，所以将其舍去

 47     }

 48 }

 49 

 50 int tuopu(int n)

 51 {

 52     stack<int> s;

 53     for(int i = 0 ; i<n ; i++){

 54         if(vis[i] && !in[i]) s.push(i);

 55     }

 56     int flag = 1;

 57     for(int i=0 ; i<cnt ; i++){

 58         if(s.empty()) return -1;

 59         if(s.size() > 1) flag = 0;

 60         int u = s.top();

 61         s.pop();

 62         for(int j=first[u] ; j!=-1 ; j=e[j].next){

 63             int v = e[j].y;

 64             in[v]--;

 65             if(!in[v]){

 66                 s.push(v);

 67             }

 68         }

 69     }

 70     if(!flag) return 0;

 71     return 1;

 72 }

 73 

 74 int main()

 75 {

 76    // freopen("a.in" , "rb" ,stdin);

 77     int n , m , a[N] , b[N];

 78     while(scanf("%d%d",&n,&m)!=EOF)

 79     {

 80         init(n);

 81         cnt = n;

 82 

 83         for(int i=0;i<m;i++){

 84             scanf("%d%s%d",&a[i],s[i],&b[i]);

 85             if(s[i][0] == '='){

 86                 Union(a[i],b[i]);

 87             }

 88         }

 89         //cout<<"cnt: "<<cnt<<endl;

 90         for(int i = 0 ; i<m ; i++){

 91             int fa1 = find_fa(a[i]);

 92             int fa2 = find_fa(b[i]);

 93             if(s[i][0] == '<'){

 94                 add_edge(fa1 , fa2);

 95             }

 96             else if(s[i][0] == '>'){

 97                 add_edge(fa2 , fa1);

 98             }

 99         }

100         int flag = tuopu(n);

101         if(flag == 1) puts("OK");

102         if(flag == -1) puts("CONFLICT");

103         if(flag == 0) puts("UNCERTAIN");

104     }

105     return 0;

106 }