问题B：万妖穴

问题 B: 万妖穴

时间限制: 1 Sec

内存限制: 64 MB

提交: 1038

解决: 264

​​提交​​​ ​​​状态​​

题目描述

“万妖穴中有若干小穴，每处小穴中皆有一只恶妖。小穴外设有封印，汝需消灭指定几处小穴中的恶妖方能解除该小穴封印，进而消灭穴中恶妖。”

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

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

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

输入

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

第一行两个正整数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

提示

拓扑排序。如果消灭 a 之后才能消灭 b，那么就有一条 a->b 的有向边。直接跑一遍模板就
好。
注意要按当前能消灭的小穴的编号从小到大进行消灭，所以要用优先队列 priority_queue。
注意 NO 的时候输出的是无法消灭的个数，而不是能消灭的个数。

#include <cstdio>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int MAXV = 1010;
vector<int> G[MAXV], topOrder;    // 邻接表、拓扑序列
int n, m, inDegree[MAXV] = {0};    // 点数、边数、入度
//拓扑排序
int topologicalSort() {
    int num = 0;    //记录加入拓扑序列的顶点数
    priority_queue<int, vector<int>, greater<int> > q;    // 小顶堆的优先队列
    for(int i = 0; i < n; i++) {
        if(inDegree[i] == 0) {
            q.push(i);      //将所有入度为0的顶点入队
        }
    }
    while(!q.empty()) {
        int u = q.top();      //取队首顶点u
        topOrder.push_back(u);
        q.pop();
        for(int i = 0; i < G[u].size(); i++) {
            int v = G[u][i];     //u的后继结点v
            inDegree[v]--;           //顶点v的入度减1
            if(inDegree[v] == 0) {   //顶点v的入度减为0则入队
                q.push(v);
            }
        }
        G[u].clear();   //清空顶点u的所有出边（如无必要可不写）
        num++;              //加入拓扑序列的顶点数加1
    }
    return num;
}

int main() {
    scanf("%d%d", &n, &m);
    for(int i = 0; i < m; i++) {
        int u, v;
        scanf("%d%d", &u, &v);
        inDegree[v]++;    // 累计入度
        G[u].push_back(v);    // 建边
    }
    int num = topologicalSort();    // 拓扑排序
    if(num == n) {
        printf("YES\n");
        for(int i = 0; i < n; i++) {
            printf("%d", topOrder[i]);
            if(i < n - 1) printf(" ");
            else printf("\n");
        }
    } else {
        printf("NO\n%d\n", n - num);    // 无法消灭的个数
    }
    return 0;
}