题目描述
众所周知, 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
解题思路
胜负关系可以构成一张有向无环图,P1 赢了 P2 等价于结点 P1 到结点 P2 有一条边,在最终的名次序列中 P1 排在 P2 的前面。题目要求名次序列的字典序最小,这个可以用优先队列解决。
因此,我们统计边的入度,把入度为 0 的边加入到优先级队列中,每次从优先级队列选取一条边,输出,修改与该边相邻点的入度,如果修改后的入度为 0 ,加入到队列中。执行完一轮 Kahn 算法,就能得到最后的结果了。
程序源码
#include<iostream>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
int main() {
std::ios::sync_with_stdio(false);
int n, m;
while (cin >> n >> m) {
priority_queue<int, vector<int>, greater<int>> q1; //优先级队列,升序排序
while (!q1.empty()) q1.pop();
int cat[550][550] = { 0 };
int rudian[550] = { 0 };
int p1, p2;
for (int i = 0; i < m; i++) { //获取输入
cin >> p1 >> p2;
if (!cat[p1][p2]) { //防止重复计数
cat[p1][p2] = 1; //打上标记
rudian[p2]++; //统计入度
}
}
for (int i = 1; i <= n; i++) { //找到入度为0的点
if (rudian[i] == 0) {
q1.push(i); //放到优先级队列中
}
}
int count = 1;
while (q1.size()) {
int current = q1.top(); //从优先级队列中取出第一个元素
if (count == n) { //输出
cout << current << endl;
break;
}
else {
cout << current << " ";
count++;
}
q1.pop(); //出队列
for (int i = 1; i <= n; i++) { //修改与当前结点指向点的入度
if (cat[current][i] == 1) {
rudian[i]--;
if (rudian[i] <= 0) { //入度为0,入队列
q1.push(i);
}
}
}
}
}
}