HDU 1811 Rank of Tetris（拓扑排序+并查集）

  1 #include<iostream>
  2 #include<algorithm>
  3 #include<cstring>
  4 #include<cstdio>
  5 #include<sstream>
  6 #include<vector>
  7 #include<stack>
  8 #include<queue>
  9 #include<cmath>
 10 #include<map>
 11 #include<set>
 12 using namespace std;
 13 typedef long long ll;
 14 typedef pair<int,int> pll;
 15 const int INF = 0x3f3f3f3f;
 16 const int maxn = 10000 + 5;
 17 
 18 int n,m;
 19 int pre_n;
 20 int p[maxn];
 21 int vis[maxn];
 22 int degree[maxn];
 23 char c[2*maxn];
 24 int u[2*maxn],v[2*maxn];
 25 bool flag1,flag2;
 26 vector<int> G[maxn];
 27 
 28 int Find(int x)
 29 {
 30     return p[x]==x?x:p[x]=Find(p[x]);
 31 }
 32 
 33 void tuopu()
 34 {
 35     memset(vis,0,sizeof(vis));
 36     for(int i=1;i<=n;i++)
 37     {
 38         int pos=-1;
 39         int num=0;
 40         for(int j=0;j<pre_n;j++)
 41         {
 42             int u=Find(j);
 43             if(vis[u]!=i && degree[u]==0)
 44             {
 45                 vis[u]=i;
 46                 pos=u;
 47                 num++;
 48             }
 49         }
 50         if(num>1)  flag1=true;
 51         if(pos==-1)  {flag2=true;return;}
 52         degree[pos]=-1;
 53         for(int j=0;j<G[pos].size();j++)
 54         {
 55             int v=G[pos][j];
 56             degree[v]--;
 57         }
 58     }
 59     return;
 60 }
 61 
 62 int main()
 63 {
 64     //freopen("in.txt","r",stdin);
 65     while(~scanf("%d%d",&n,&m))
 66     {
 67         pre_n=n;
 68         for(int i=0;i<n;i++)  G[i].clear();
 69         memset(degree,0,sizeof(degree));
 70         for(int i=0;i<n;i++)  p[i]=i;
 71 
 72         for(int i=0;i<m;i++)
 73         {
 74             cin>>u[i]>>c[i]>>v[i];
 75             if(c[i]==‘=‘)
 76             {
 77                 int x=Find(u[i]);
 78                 int y=Find(v[i]);
 79                 if(x!=y)
 80                 {
 81                     p[x]=y;
 82                     n--;
 83                 }
 84             }
 85         }
 86 
 87         for(int i=0;i<m;i++)
 88         {
 89             int x=Find(u[i]);
 90             int y=Find(v[i]);
 91             if(c[i]==‘>‘)
 92             {
 93                 G[x].push_back(y);
 94                 degree[y]++;
 95             }
 96             else if(c[i]==‘<‘)
 97             {
 98                 G[y].push_back(x);
 99                 degree[x]++;
100             }
101         }
102 
103         flag1=flag2=false;
104         tuopu();
105         if(flag2)  puts("CONFLICT");
106         else if(flag1)  puts("UNCERTAIN");
107         else puts("OK");
108     }
109     return 0;
110 }