拓扑排序 - 并查集 - Rank of Tetris

OK
CONFLICT
UNCERTAIN


--------------------------------------------------我是分割线^_^-------------------------------------------------------------



这道题也是WA不少次了，我是事后看题解才反应过来，很无语的一道题，要先全部接受输入把等号处理
掉，我之前一边输入一边处理等号，TLE到凌晨才发现= =，主要就是先用并查集把相等的元素全部合并
然后当作一个数处理，然后就是一个拓扑排序。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<vector>
#include<queue>
using namespace std;

#define Int __int64
#define INF 0x3f3f3f3f

const int MAXN = 22222;
int team[MAXN];
int head[MAXN];
int point[MAXN];
int nxt[MAXN];
int edgecnt;
int ingrade[MAXN];
int n, m;
int road[MAXN];
int sum;
int A[MAXN], B[MAXN];
char op[MAXN];

void ini() {
    memset(head, -1, sizeof(head));
    memset(ingrade, 0, sizeof(ingrade));
    edgecnt = 0;
    sum = n;
    for (int i = 0; i < n; i++) {
        road[i] = i;
    }
}

void Add_Edge(int u, int v) {
    nxt[edgecnt] = head[u];
    point[edgecnt] = v;
    head[u] = edgecnt++;
}

int FindRoot(int rt) {
    return road[rt] == rt ? rt : (road[rt] = FindRoot(road[rt]));
}

int main()
{
    //freopen("input.txt", "r", stdin);
    while (scanf("%d %d\n", &n, &m) != EOF) {
        ini();
        for (int i = 0; i < m; i++) {
            scanf("%d %c %d", &A[i], &op[i], &B[i]);
            if (op[i] == ‘=‘) {
                int root1 = FindRoot(A[i]);
                int root2 = FindRoot(B[i]);
                if (root1 != root2) {
                    road[root2] = root1;
                    sum--;//记得合并之后减一，这才是当前的集合个数
                }
            }
        }
        for (int i = 0; i < m; i++) {
            if (op[i] == ‘=‘) {
                continue;
            }
            int a = A[i];
            int b = B[i];
            a = FindRoot(a);
            b = FindRoot(b);
            if (op[i] == ‘>‘) {
                Add_Edge(a, b);
                ingrade[b]++;
            } else {
                Add_Edge(b, a);
                ingrade[a]++;
            }
        }
        queue<int>q;
        while (!q.empty()) q.pop();

        for (int i = 0; i < n; i++) {
            if (ingrade[i] == 0 && FindRoot(i) == i) q.push(i);//注意此时的road[i]的根节点可能未更新，因此查询顺便更新
        }
        bool flag = false;
        while (!q.empty()) {
            if (q.size() > 1) flag = true;//如果入度为零的点不止一个，那肯定有多组解，答案不明
            int now = q.front();
            q.pop();
            sum--;//这里的处理就是为了下面判断还有木有边剩下来，如果有，肯定就是有环啦！
            for (int i = head[now]; i != -1; i = nxt[i]) {
                int v = point[i];
                ingrade[v]--;
                if (!ingrade[v]) q.push(v);
            }
        }
        if (sum > 0) {
            printf("CONFLICT\n");
        } else if (flag) {
            printf("UNCERTAIN\n");
        } else {
            printf("OK\n");
        }

    }
    return 0;
}