[abc] D - Restricted Permutation 拓扑排序

前言

很裸的一个 拓扑排序题
传送门 :

思路

一开始考虑了一下感觉可以直接走拓扑序,

不过需要使用 $stack$ 来替换队列,并且还需要一开始逆序求 入度为0的点,后面只过了一半的样例,还是我太年轻了

其实对于字典序的输出,我们可以使用 $小根堆$ 来很好的维护 因此这题就是一个简单题了

CODE

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
const int N  = 2e5+10 , M = N*2;
int n,m;
int h[N],e[M],ne[M],idx;
int in[N];
int ans[N],id;
void add(int a,int b)
{
    e[idx] = b,ne[idx] = h[a],h[a] =idx++;
}

bool topsort()
{
    int num = 0;

    priority_queue<int,vector<int>,greater<int>> q;

    for(int i=n;i>=1;i--)
    {
        if(!in[i])
            q.push(i);
    }
    while(!q.empty())
    {
        num++;
        int t = q.top();
        ans[++id] = t;
        q.pop();

        for(int i=h[t];~i;i=ne[i])
        {
            int j = e[i];
            --in[j];
            if(!in[j])
            q.push(j);
        }
    }
    if(num == n)
    return true;
    else
    return false;
}
void solve()
{
    memset(h,-1,sizeof h);
    cin>>n>>m;
    while(m -- )
    {
        int a,b;cin>>a>>b;
        add(a,b);
        in[b]++;
    }
    if(topsort())
    {
        for(int i=1;i<=id;i++)
        cout<<ans[i]<<" " ;
    }
    else
    cout<<-1<<endl;

}

int main()
{
    ios::sync_with_stdio(false);
    solve();
    return 0;
}