貓貓向前衝(拓撲排序)

問題描述

衆所周知, 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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章