拓扑排序模板

题目链接: https://vjudge.net/problem/Gym-100792K
大的尽量往后放
普通写法tle,用优先队列优化。不是字典序输出,正向拓扑无法解决,反过来即可。

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+5;
int ans[maxn];
vector <int> g[maxn];
int n,m;
int in[maxn];
void topsort()
{
    priority_queue <int> q;
    for(int i=1; i<=n; i++)
    {
        if(!in[i])
            q.push(i);
    }
    int cnt=n-1;
    while(!q.empty())
    {
        int u=q.top();
        q.pop();
        ans[cnt--]=u;
        in[u]=-1;
        int l=g[u].size();
        for(int i=0; i<l; i++)
        {
            if(in[g[u][i]]>0)
            {
                in[g[u][i]]--;
                if(in[g[u][i]]==0)
                q.push(g[u][i]);
            }

        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0; i<m; i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        in[a]++;
        g[b].push_back(a);
    }
    topsort();
    for(int i=0; i<n; i++)
    {
        if(i)
            printf(" ");
        printf("%d",ans[i]);
    }
    return 0;
}

hdu 1285
题意:给你n个数,m行输入,每行输入的两个数a,b代表a<b,输出大小关系,要求字典序最小。注意题目中已经说一定有答案,即没有自环。
坑点:拓扑排序并且字典序最小,紫书上的模板不能输出字典序最小。
常规写法:

#include <bits/stdc++.h>

using namespace std;
const int maxn=505;
int g[maxn][maxn],in[maxn],ans[maxn];
int n,m;
void topsort()
{
    for(int i=1; i<=n; i++)
    {
        int k=1;
        while(in[k]!=0)//找到最小的数
        {
            k++;
        }
        ans[i]=k;
        in[k]=-1;
        for(int j=1; j<=n; j++)//修改
        {
            if(g[k][j])
            {
                in[j]--;
            }
        }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(g,0,sizeof(g));
        memset(in,0,sizeof(in));
        memset(ans,0,sizeof(ans));
        for(int i=0; i<m; i++)
        {

            int a,b;
            scanf("%d%d",&a,&b);
            if(!g[a][b])//防止有重边
            {
                in[b]++;
            }
            g[a][b]=1;
        }
        topsort();
        for(int i=1; i<=n; i++)
        {
            if(i!=1)
                printf(" ");
            printf("%d",ans[i]);
        }
        printf("\n");
    }

    return 0;
}

优先队列优化版本:
最后判断一下cnt的大小即可知道是否有自环

#include <bits/stdc++.h>

using namespace std;
const int maxn=505;
int g[maxn][maxn],in[maxn],ans[maxn];
int n,m;
void topsort()
{
    priority_queue <int,vector <int>,greater<int> >q;
    for(int i=1; i<=n; i++)
    {
        if(!in[i])
            q.push(i);
    }
    int cnt=0;
    while(!q.empty())
    {
        int x=q.top();
        q.pop();
        ans[++cnt]=x;
        for(int i=1; i<=n; i++)
        {
            if(g[x][i])
            {
                in[i]--;
                if(in[i]==0)
                    q.push(i);
            }
        }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(g,0,sizeof(g));
        memset(in,0,sizeof(in));
        memset(ans,0,sizeof(ans));
        for(int i=0; i<m; i++)
        {

            int a,b;
            scanf("%d%d",&a,&b);
            if(!g[a][b])//防止有重边
            {
                in[b]++;
            }
            g[a][b]=1;
        }
        topsort();
        for(int i=1; i<=n; i++)
        {
            if(i!=1)
                printf(" ");
            printf("%d",ans[i]);
        }
        printf("\n");
    }

    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值