猫猫向前冲(拓扑排序)

问题描述

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