HihoCoder - 1174 拓扑排序·一

由于今天上课的老师讲的特别无聊，小Hi和小Ho偷偷地聊了起来。

小Ho：小Hi，你这学期有选什么课么？

小Hi：挺多的，比如XXX1,XXX2还有XXX3。本来想选YYY2的，但是好像没有先选过YYY1，不能选YYY2。

小Ho：先修课程真是个麻烦的东西呢。

小Hi：没错呢。好多课程都有先修课程，每次选课之前都得先查查有没有先修。教务公布的先修课程记录都是好多年前的，不但有重复的信息，好像很多都不正确了。

小Ho：课程太多了，教务也没法整理吧。他们也没法一个一个确认有没有写错。

小Hi：这不正是轮到小Ho你出马的时候了么！

小Ho：哎？？

我们都知道大学的课程是可以自己选择的，每一个学期可以自由选择打算学习的课程。唯一限制我们选课是一些课程之间的顺序关系：有的难度很大的课程可能会有一些前置课程的要求。比如课程A是课程B的前置课程，则要求先学习完A课程，才可以选择B课程。大学的教务收集了所有课程的顺序关系，但由于系统故障，可能有一些信息出现了错误。现在小Ho把信息都告诉你，请你帮小Ho判断一下这些信息是否有误。错误的信息主要是指出现了"课程A是课程B的前置课程，同时课程B也是课程A的前置课程"这样的情况。当然"课程A是课程B的前置课程，课程B是课程C的前置课程，课程C是课程A的前置课程"这类也是错误的。

提示：拓扑排序

输入

第1行：1个整数T，表示数据的组数T(1 <= T <= 5)
接下来T组数据按照以下格式：
第1行：2个整数，N,M。N表示课程总数量，课程编号为1..N。M表示顺序关系的数量。1 <= N <= 100,000. 1 <= M <= 500,000
第2..M+1行：每行2个整数，A,B。表示课程A是课程B的前置课程。

输出

第1..T行：每行1个字符串，若该组信息无误，输出"Correct"，若该组信息有误，输出"Wrong"。

样例输入

2
2 2
1 2
2 1
3 2
1 2
1 3

样例输出

Wrong
Correct

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

vector<int>to[100000+8];
queue<int>q;
priority_queue<int, vector<int>, greater<int> >you;

int n, m, t, in[1000000+8], sign[1000000+8], du[1000000+8], si[1000000+8];

int get(int miao)
{
    for(int i = 0; i <= miao; i++)
    {
        du[i] = in[i];
        si[i] = sign[i];
    }
    int num = miao;
    while(!q.empty())
    {
        int t = q.front();
        num--;
        q.pop();
        for(int i = 0; i<to[t].size(); i++)
        {
            du[to[t][i]]--;
            if(!du[to[t][i]] && !si[to[t][i]])
            {
                q.push(to[t][i]);
                si[to[t][i]] = 1;
            }
        }
    }
    return num;
}

int main()
{
    int a, b;
    for(scanf("%d", &t);t--;)
    {
        memset(in, 0, sizeof(in));
        memset(sign, 0, sizeof(sign));
        for(int i = 0; i <= n; i++)to[i].clear();
        scanf("%d%d", &n, &m);
        for(int i = 0; i<m; i++)
        {
            scanf("%d%d", &a, &b);
            to[a].push_back(b);
            in[b]++;
        }
        for(int i = 1; i <= n; i++)
        {
            if(!in[i] && !sign[i])
            {
                q.push(i);
                you.push(i);
                sign[i] = 1;
            }
        }
        if(get(n) == 0)
            printf("Correct\n");
        else printf("Wrong\n");
    }
    return 0;
}