問題描述
衆所周知, 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;
}