题目描述
众所周知, 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算法即可。方法是存图的同时使用inDeg数组记录每个点入度数,入度为0的点是最厉害的,因为没有点指向它,首先遍历inDeg将他们取出压入小根堆。然后依次从堆顶读取点并将其压入输出向量ans,同时遍历其指向的点,将其入度-1,一旦有一个点入度为0,立即将其压入小根堆。循环的终止条件是堆空。为保证是字典序最小的名次序列,我们使用小根堆存待排序点。
本题需注意的是:
- 有多组输入。一开始没有好好读题WA了。
- 邻接矩阵存图会TLE,长长记性,尽量少用。
碎碎念
因为priority_queue默认是大根堆,声明普通类型小根堆使用如下:
priority_queue< int, vector< int >, greater< int > > q;
但倘若需要使用结构体就在结构体内外定义比较函数即可。
代码
#define _CRT_SECURE_NO_WARNINGS
#define _ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include <bits/stdc++.h>
//#include <iostream>
//#include<algorithm>
using namespace std;
int N, M, a, b;
int inDeg[509];//入度
//bool cat[509][509];//存图 邻接矩阵存图TLE了。。。
vector<int>G[509];
int main()
{
while (cin>>N>>M)
{
memset(inDeg, 0, sizeof(inDeg));
memset(G, 0, sizeof(G));
while (M--)
{
cin >> a >> b;
G[a].push_back(b);
inDeg[b]++;
}
priority_queue< int, vector< int >, greater< int > > q;//声明小根堆
vector<int> ans;
for (int i = 1; i <= N; i++)
if (inDeg[i] == 0)q.push(i);
while (!q.empty())
{//Kahn算法
int u = q.top(); q.pop();
ans.push_back(u);
for (int i = 0; i<G[u].size(); i++)
{
int v = G[u][i];
inDeg[v]--;
if (inDeg[v] == 0)q.push(v);
}
}
for (int i = 0; i < N; i++)
{
if (i == 0)cout << ans[0];
else cout << ' ' << ans[i];
}
cout << endl;
}
return 0;
}