27734: 万妖穴

题目描述

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

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

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

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

输入

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

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

样例输出

YES0 2 1

提示

来源

Shoutmon

拓扑排序

#include<iostream>
#include<vector>
#include<functional>
#include<queue>
#include<algorithm>

using namespace std;

const int MAX_N=1001;
vector <int> G[MAX_N],topOrder;

int vertex,edge,inDegree[MAX_N]={0};
int num;

bool topologicalsort(){
	num=0;
	priority_queue<int,vector<int>,greater<int> > q;
	for(int i=0;i<vertex;i++){
		if(inDegree[i]==0)
			q.push(i);
	}
	while(!q.empty()){
		int u=q.top();
		topOrder.push_back(u);
		q.pop();
		for(int i=0;i<G[u].size();i++){
			int v=G[u][i];
			inDegree[v]--;
			if(inDegree[v]==0)
				q.push(v);
		}
		num++;
	}
	if(num==vertex)	return true;
	else	return false;
}



int main(){
	cin>>vertex>>edge;
	for(int i=0;i<edge;i++){
		int u,v;
		cin>>u>>v;
		inDegree[v]++;
		G[u].push_back(v);
	}
	if(topologicalsort()){
		cout<<"YES"<<endl;
		for(int i=0;i<vertex;i++){
			cout<<topOrder[i];
			if(i<vertex-1){
				cout<<" ";
			}
			else
				cout<<endl;
		}
	}
	else{
		cout<<"NO"<<endl;
		cout<<vertex-num<<endl;
	}
}