题目原文
众所周知, 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
解题思路
由于排名有先后性,且要求输出排名,故考虑用拓扑排序实现。
由于符合条件的输出有多种,题目要求输出字典序最小的答案,因此将拓扑排序中的队列改为优先队列(最大优先队列,由于要从小到大排序,因此存入时存入原值的相反数,输出时,输出值乘以-1还原),并在每次数据出队列时,用score[]数组记录每只猫猫的编号。
最后按输出格式要求输出score[].
实现代码
#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
const int maxn=510;
int n,m;
bool cat[maxn][maxn];
int d[maxn];
int score[maxn];
int cnt;
void toposort(){
cnt=0;
priority_queue<int>q;
for(int i=1;i<=n;i++)
{
if(d[i]==0)
q.push(-i);
}
while(!q.empty())
{
int x=-q.top();
q.pop();
score[++cnt]=x;
//判断删除节点能到达的点的
for(int y=1;y<=n;y++)
{
if(cat[x][y])
{
d[y]--;
if(d[y]==0)
{
q.push(-y);
}
}
}
}
}
int main(){
while(scanf("%d %d",&n,&m)!=EOF)
{
memset(cat,0,sizeof(cat));
memset(d,0,sizeof(d));
int x,y;
for(int i=1;i<=m;i++)
{
scanf("%d %d",&x,&y);
if(cat[x][y]==1)
continue;
cat[x][y]=1;
d[y]++;
}
toposort();
bool mark=0;
for(int i=1;i<=n;i++)
{
if(!mark)
{
printf("%d",score[i]);
mark=1;
}
else
printf(" %d",score[i]);
}
printf("\n");
}
return 0;
}
小结
在队列值需要排序时,优先考虑用堆(优先队列)解决,比手写比较方法简单快捷。