Tetris Rank Determination using Union-Find and Topological Sort (HDU 1811)

1 /*
2 题意：这些信息可能有三种情况，分别是"A > B","A = B","A < B"，分别表示A的Rating高于B,等于B,小于B。
3
4 现在Lele并不是让你来帮他制作这个高手榜，他只是想知道，根据这些信息是否能够确定出这个高手榜，是的话就输出"OK"。
5 否则就请你判断出错的原因，到底是因为信息不完全（输出"UNCERTAIN"），还是因为这些信息中包含冲突（输出"CONFLICT"）。
6 注意，如果信息中同时包含冲突且信息不完全，就输出"CONFLICT"。
7
8 思路: 因为小于关系和大于关系可以转换一下位置！ 这里的问题就在与如何正确的处理相等的关系！
9 如果没有相等的关系，一个拓扑排序算法就可以搞定了！ 既然元素相等，那么我们取相等元素中的某一个
10 数来表示每一个数不是也行吗！？对，没错，用这个数来代替所有与之相等元素的数表示 '<'关系！ 也就是
11 转换成集合之间的关系的处理！ 将每一个相等的元素集合看成一个点，这个点的代表就是集合的父亲节点！
12
13 那么如何来得到这个数呢？并查集最适合不过了！我们将相等的元素放入集合中！
14 当 a<b时，通过getFather(a) < getFather(b)来处里a<b的关系，这里用邻接表进行处理！
15 */
16 #include<iostream>
17 #include<cstring>
18 #include<vector>
19 #include<stack>
20 using namespace std;
21 int f[10005];
22 int rank[10005];
23 int n, m;
24 int getFather(int x){
25 return x==f[x] ? x : f[x]=getFather(f[x]);
26 }
27
28 int Union(int a, int b){
29 int fa=getFather(a), fb=getFather(b);
30 if(fa!=fb){
31 if(rank[fa]>rank[fb]){
32 f[fb]=fa;
33 rank[fa]+=rank[fb]+1;
34 }
35 else{
36 f[fa]=fb;
37 rank[fb]+=rank[fa]+1;
38 }
39 return 1;
40 }
41 return 0;
42 }
43
44 int in[10005];
45 int A[10005], B[10005];
46 char ch[10005];
47 vector<int>vx[10005];
48 int conflict, uncertain;
49 int sum;
50
51 /*void topoSort(){
52 for(int j=1; j<=sum; ++j){
53 int p=0, cnt=0;
54 for(int i=1; i<=n; ++i)
55 if(f[i]==i && in[i]==0){//f[i]==i表明 i是这个相等集合的代表元素，也就是这个集合所有元素的父节点
56 p=i;
57 ++cnt;
58 }
59 if(cnt==0){
60 conflict=1;
61 return;
62 }
63 if(cnt>1)
64 uncertain=1;
65 int len=vx[p].size();
66 for(int i=0; i<len; ++i)
67 --in[vx[p][i]];
68 in[p]=-1;
69 }
70 }*/
71
72 stack<int>ss;
73
74 void topoSort(){
75 for(int i=1; i<=n; ++i)
76 if(f[i]==i && in[i]==0)//f[i]==i表明 i是这个相等集合的代表元素，也就是这个集合所有元素的父节点
77 ss.push(i);
78 if(ss.size()==0 && sum)
79 conflict=1;
80 while(!ss.empty()){
81 int cnt=ss.size();
82 int p=ss.top();
83 --sum;//表示剩余多少个节点没有排序！
84 ss.pop();
85
86 if(cnt>1)
87 uncertain=1;
88 int len=vx[p].size();
89 for(int i=0; i<len; ++i)
90 if(--in[vx[p][i]]==0)
91 ss.push(vx[p][i]);
92 if(ss.size()==0 && sum)
93 conflict=1;
94 }
95 }
96
97 int main(){
98 while(cin>>n>>m){
99 for(int i=1; i<=n; ++i)
100 f[i]=i;
101 for(int i=1; i<=m; ++i){
102 scanf("%d %c %d", &A[i], &ch[i], &B[i]);
103 ++A[i];
104 ++B[i];
105 if(ch[i]=='=')
106 Union(A[i], B[i]);
107 }
108 sum=0;
109 for(int i=1; i<=n; ++i)
110 if(f[i]==i) ++sum;
111 for(int i=1; i<=m; ++i){
112 int fa=getFather(A[i]), fb=getFather(B[i]);//将每一个相等的元素集合看成一个点，这个点的代表就是其父亲节点
113 if(ch[i]=='<'){
114 vx[fa].push_back(fb);
115 ++in[fb];
116 }
117 else if(ch[i]=='>'){
118 vx[fb].push_back(fa);
119 ++in[fa];
120 }
121 }
122
123 conflict=uncertain=0;
124 topoSort();
125 if(conflict)
126 cout<<"CONFLICT"<<endl;
127 else if(uncertain)
128 cout<<"UNCERTAIN"<<endl;
129 else cout<<"OK"<<endl;
130 for(int i=1; i<=n; ++i)
131 vx[i].clear();
132
133 memset(rank, 0, sizeof(int)*(n+1));
134 memset(in, 0, sizeof(int)*(n+1));
135 while(!ss.empty())
136 ss.pop();
137 }
138 }