问题 B: 万妖穴

时间限制: 1 Sec内存限制: 64 MB提交: 1020解决: 255
提交状态
题目描述
“万妖穴中有若干小穴，每处小穴中皆有一只恶妖。小穴外设有封印，汝需消灭指定几处小穴中的恶妖方能解除该小穴封印，进而消灭穴中恶妖。”

“此处石壁所刻便是万妖穴中各穴的封印解除关系。”

“万妖穴中或有部分小穴未设封印，汝可以此作为突破口。”

“吾将根据石壁所示，将各穴消灭顺序给予汝，汝自珍重。”

输入
每个输入文件中一组数据。

第一行两个正整数N、M（1<=N<=1000，0<=M<=N*(N-1)），表示万妖穴中小穴的个数及封印解除关系的条数。

接下来M行，每行两个整数a、b，表示小穴b封印的解除依赖于消灭小穴a中的恶妖。假设N处小穴的编号分别是0~N-1。数据保证a不等于b，且每组封印解除关系a b最多出现一次。

输出
如果能够消灭所有小穴，那么输出一行YES，并在第二行给出用空格隔开的N处小穴的消灭顺序（此处规定，总是消灭当前能消灭的编号最小的小穴）；如果不能消灭所有小穴，那么输出一行NO，并在第二行给出无法解除封印的小穴个数。行末均不允许输出多余的空格。

样例输入和输出那里贴不了多个case（提交的程序会强行测试sample），所以直接放这里吧
input:
// case 1
3 3
0 1
0 2
2 1
// case 2
4 5
0 1
0 2
0 3
2 1
1 2
output:
// case 1
YES
0 2 1
// case 2
NO
2
样例输入
3 3
0 1
0 2
2 1
样例输出
YES
0 2 1
提示
提交状态

考察拓扑排序，学过编过拓扑排序应该没什么难度

#include<iostream>
#include<vector>

using namespace std;
int arc[1005][1005] = {};
int indegree[1005] = {};
int main()
{
    int N, M;
    cin >> N >> M;
    while (M--)
    {
        int a, b;
        cin >> a >> b;
        arc[a][b] = 1;
        indegree[b]++;
    }
    vector<int> re;
    vector<bool> visited(1005,false);
    while (re.size() != N)//拓扑排序
    {
        int v = N;
        for (int t = 0;t < N;t++)
            if (!visited[t]&&!indegree[t]) { v = t;break; }
        if (v == N) break;
        re.push_back(v);
        visited[v] = true;
        for (int t = 0;t < N;t++)
            if (!visited[t] && arc[v][t] != 0) indegree[t]--;
    }
    if (re.size() != N) cout << "NO" << endl << N - re.size() << endl;
    else
    {
        cout << "YES" << endl;
        cout << re[0];
        for (int t = 1;t < N;t++)
            cout << " " << re[t];
        cout << endl;

    }
}