0
点赞
收藏
分享

微信扫一扫

拓扑排序:奖金

Python百事通 2022-03-12 阅读 41

原题链接:https://www.acwing.com/problem/content/description/1194/
在这里插入图片描述
在这里插入图片描述

/*
    反向建图,建立b->a的图,然后求一篇拓扑排序
    假设有虚拟源点,刚开始从根节点向下扩展,每个点都是100
    然后顺着拓扑序,每个点都需要在父节点的基础上加一,
    一个点有可能有多个父节点,要维护一个最大值。
*/

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 10010, M = 20010;

int n, m;
int h[N], e[M], ne[M], idx;
int q[N];
int d[N];
int dist[N];

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

bool topsort()
{
    int hh = 0, tt = -1;
    for (int i = 1; i <= n; i ++ )
        if (!d[i])
            q[ ++ tt] = i;

    while (hh <= tt)
    {
        int t = q[hh ++ ];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if ( -- d[j] == 0)
                q[ ++ tt] = j;
        }
    }

    return tt == n - 1;
}

int main()
{
    scanf("%d%d", &n, &m);
    memset(h, -1, sizeof h);
    while (m -- )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        add(b, a);
        d[a] ++ ;
    }

    if (!topsort()) puts("Poor Xed");
    else
    {
        for (int i = 1; i <= n; i ++ ) dist[i] = 100;
        // 从头到尾遍历队列中所有点
        for (int i = 0; i < n; i ++ )
        {
            int j = q[i];
            // 对于每个点指向的点,维护指向的点的最大值。
            for (int k = h[j]; ~k; k = ne[k])
                dist[e[k]] = max(dist[e[k]], dist[j] + 1);
        }

        int res = 0;
        for (int i = 1; i <= n; i ++ ) res += dist[i];

        printf("%d\n", res);
    }

    return 0;
}

举报

相关推荐

0 条评论