问题描述
众所周知, 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
解题思路
基本的拓扑排序问题,就是在存储结果时,使用小根堆优先队列即可。注意,多组数据!
这个题好像不支持priority_queue<int, vector<int>,greater<int> >q;我就将负数丢到队列中了,这样也可以实现小根堆的效果。
完整代码
//#pragma GCC optimize(2)
//#pragma G++ optimize(2)
//#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <climits>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
const int maxn=500+10;
struct node
{
int to,next;
};
node edge[maxn*2];
int n,m,indegree[maxn],head[maxn],ans[maxn],cnt,cnt2;
void add(int x,int y)
{
cnt++;
edge[cnt].to=y;
edge[cnt].next=head[x];
head[x]=cnt;
}
bool toposort()
{
priority_queue<int>q;
while(!q.empty()) q.pop();
for (int i=1; i<=n; i++)
if(indegree[i]==0)
q.push(-i);
while(!q.empty()){
int u=-q.top(); q.pop();
ans[++cnt2]=u;
for (int i=head[u]; i; i=edge[i].next)
if(--indegree[edge[i].to]==0)
q.push(-edge[i].to);
}
return cnt2==n;
}
int getint()
{
int x=0,s=1;
char ch=' ';
while(ch<'0' || ch>'9')
{
ch=getchar();
if(ch=='-') s=-1;
}
while(ch>='0' && ch<='9')
{
x=x*10+ch-'0';
ch=getchar();
}
return x*s;
}
int main()
{
while(cin>>n>>m){
for (int i=1; i<=m; i++){
int a=getint(); int b=getint();
add(a,b);
indegree[b]++;
}
toposort();
for (int i=1; i<n; i++)
printf("%d ",ans[i]);
printf("%d\n",ans[n]);
cnt=0; cnt2=0;
memset(head,0,sizeof(head));
memset(indegree,0,sizeof(indegree));
}
return 0;
}