【ACM】- HDU-2647 Reward 【拓扑排序】

题目链接

题目分析

拓扑排序失败则输出-1；否则输出总花费

解题思路

逆拓扑排序：反向保存图信息
每次删边时更新子节点的value值，取 max(父结点值 + 1, 原始值)

---

AC程序（C++）

/**********************************
*@ID: 3stone
*@ACM: HDU-2647 Reward
*@Time: 18/9/10
*@IDE: VSCode + clang++
***********************************/
#include<cstdio>
#include<algorithm>
#include<queue>
#include<iostream>

using namespace std;
const int maxn = 10010;

int N, M;

int value[maxn]; //结点红包值
int cnt[maxn]; //保存结点的入度
vector<int> Adj[maxn]; //邻接表

//拓扑排序
void topsort() {
    priority_queue<int> Q; //用堆实现，每次先输出编号最小的
    for(int i = 1; i <= N; i++) { //寻找入度为0的顶点
        if(cnt[i] == 0)
            Q.push(i);
    }

    int num = 0;
    while(!Q.empty()) {
        num++;
        int v = Q.top();
        Q.pop();

        for(int i = 0; i < Adj[v].size(); i++){
            value[Adj[v][i]] = max(value[Adj[v][i]], value[v] + 1); //更新红包
            //删除边，入度-1,判断
            if(--cnt[Adj[v][i]] == 0)
                Q.push(Adj[v][i]);
        }
    }//while

    if(num != N) printf("-1\n");
    else {
        int sum = 0;
        for(int i = 1; i <= N; i++) sum += value[i];
        printf("%d\n", sum);
    }

}

int main() {
    int a, b;
    while(scanf("%d %d", &N, &M) != EOF) {
        //初始化 邻接表
        for(int i = 0; i < maxn; i++) Adj[i].clear();
        fill(cnt, cnt + maxn, 0);
        fill(value, value + maxn, 888);

        //输入结点偏序
        for(int i = 1; i <= M; i++) { 
            scanf("%d %d", &a, &b);
            Adj[b].push_back(a); //逆拓扑排序：反向存储
            cnt[a]++;
        }

        topsort(); //拓扑排序
    }

    system("pause");
    return 0;
}