首页 > 代码库 > hdu 1811Rank of Tetris (并查集 + 拓扑排序)

hdu 1811Rank of Tetris (并查集 + 拓扑排序)

  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 }