程序设计思维 B - 猫猫向前冲（拓扑排序、Kahn算法）

题目

众所周知， TT 是一位重度爱猫人士，他有一只神奇的魔法猫。
有一天，TT 在 B 站上观看猫猫的比赛。一共有 N 只猫猫，编号依次为1，2，3，…，N进行比赛。比赛结束后，Up 主会为所有的猫猫从前到后依次排名并发放爱吃的小鱼干。不幸的是，此时 TT 的电子设备遭到了宇宙射线的降智打击，一下子都连不上网了，自然也看不到最后的颁奖典礼。
不幸中的万幸，TT 的魔法猫将每场比赛的结果都记录了下来，现在他想编程序确定字典序最小的名次序列，请你帮帮他。

Input
输入有若干组，每组中的第一行为二个数N（1<=N<=500），M；其中N表示猫猫的个数，M表示接着有M行的输入数据。接下来的M行数据中，每行也有两个整数P1，P2表示即编号为 P1 的猫猫赢了编号为 P2 的猫猫。

Output
给出一个符合要求的排名。输出时猫猫的编号之间有空格，最后一名后面没有空格!
其他说明：符合条件的排名可能不是唯一的，此时要求输出时编号小的队伍在前；输入数据保证是正确的，即输入数据确保一定能有一个符合要求的排名。

Sample Input
4 3
1 2
2 3
4 3

Sample Output
1 2 4 3

思路

一、Kahn算法
Kahn算法是解决拓扑排序的一种算法。
要用Kahn算法求图的字典序最小的拓扑序列，需要用到1个最小堆（优先队列）、1个线性表：
最小堆q：存放度为0的点，且每次取出当中字典序最小的点
线性表v：存放拓扑序列
Kahn算法：
（1）将所有度为0的点放入q中
（2）从q中取出一个点i并放入v中，然后将所有从点i出发的边（i, j）删除，并将点j的度-1，如果点j的度减到0，则将点j放入q中
（3）过程（2）重复至q为空，最后的v即为所求的字典序最小的拓扑序列
二、解题
1.构造图
若编号为 P1 的猫猫赢了编号为 P2 的猫猫，则将构造一条边（P1, P2），并将P2的度+1。
2.求图的字典序最小的拓扑序列
可以直接用Kahn算法求图的字典序最小的拓扑序列。

代码

#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
#include <list>
#include <algorithm>

#define MAX_V 600

using namespace std;

int N, M;
int P1, P2;

struct cmp{
	bool operator()(const int& a, const int& b)
	{
		return a > b;
	}
};

int degree[MAX_V]; // degree[0]不用
vector<int> v;
priority_queue<int, vector<int>, cmp> q;
list<int> lis[MAX_V];

void init()
{
	memset(degree, 0, sizeof(degree));
	for (int i = 1; i <= N; i++) lis[i].clear();
	while (!q.empty()) q.pop();
	v.clear();
}

int main()
{
	//cin >> N >> M
	while ((scanf("%d %d", &N, &M)) != EOF) {

		init();

		for (int i = 0; i < M; i++) {
			scanf("%d %d", &P1, &P2);
			//cin >> P1 >> P2;
			degree[P2]++;
			lis[P1].push_back(P2);
		}

		for (int i = 1; i <= N; i++) if (degree[i] == 0) q.push(i);

		while (!q.empty()) {
			int p = q.top();
			v.push_back(p);
			q.pop();

			for (list<int>::iterator it = lis[p].begin(); it != lis[p].end(); it++) {
				degree[*it]--;
				if (degree[*it] == 0) q.push(*it);
			}
		}

		for (vector<int>::iterator it = v.begin(); it != v.end();) {
			cout << *it;
			it++;
			if (it != v.end()) cout << " ";
			else cout << endl;
		}

	}
	
	return 0;
}