hdu1811 Rank of Tetris 并查集+拓扑排序

 #include <stdio.h>
 #include <string.h>
 #include <vector>
 #include <queue>
 using namespace std;
 struct node//边
 {
     int a, b;//顶点
     char ch;//运算符
 }c[];
 vector<int>map[];//map数组存贮邻接表 (大佬都是这么开数组的)
 int n, m, sum, in[], fa[];//in数组表示入度,fa[i]表示顶点i所在集合的根节点  

 int find(int x)//查找根节点
 {
     if (fa[x] != x) fa[x] = find(fa[x]);
     return fa[x];
 }

 bool comb(int x, int y)//合并集合
 {
     x = find(x);
     y = find(y);
     if (x == y)
         return false;
     else
     {
         fa[y] = x;
         return true;
     }
 }

 void init()//初始化
 {
     for (int i = ; i<n; i++)
         fa[i] = i;
 }

 void top_sort()//queue实现拓扑排序
 {
     queue<int>s;
     int flag = ;
     for (int i = ; i<n; i++)
     {
         //找到入度为零的切祖宗为自己的加入到队列中
         if (in[i] ==  && fa[i] == i)
             s.push(i);
     }
     while (!s.empty())
     {
         if (s.size() > )//即使发现信息不完整也要继续运行下去，因为如果信息同时不完整和冲突都是CONFLICT
             flag = ;
         int pos = s.front();
         s.pop(), sum--;    //记录下运行的次数
         for (int i = ; i<map[pos].size(); i++)
         {
             in[map[pos][i]]--;
             if (in[map[pos][i]] == )
                 s.push(map[pos][i]);
         }
     }
     if (sum>) printf("CONFLICT\n");    //冲突，即有多个入度为零且祖宗为自己的出现
     else if (flag) printf("UNCERTAIN\n");
     else printf("OK\n");
 }

 int main()
 {
     while (scanf("%d %d", &n, &m) != EOF)
     {
         sum = n;
         init();
         memset(map, , sizeof(map));
         memset(in, , sizeof(in));
         for (int i = ; i<m; i++)
         {
             scanf("%d %c %d", &c[i].a, &c[i].ch, &c[i].b);
             //如果相等，就合并为同一个集合
             if (c[i].ch == '=')
             {
                 if (comb(c[i].a, c[i].b))
                     sum--;
             }
         }
         for (int i = ; i<m; i++)
         {
             if (c[i].ch == '=')
                 continue;
             int x = find(c[i].a);
             int y = find(c[i].b);
             if (c[i].ch == '>')
             {
                 map[x].push_back(y);
                 in[y]++;
             }
             else
             {
                 map[y].push_back(x);
                 in[x]++;
             }
         }
         top_sort();
     }
     //system("pause");
     return ;
 }