拓扑排序实现【636.旅行计划】

-兮

于 2022-03-08 16:05:00 发布

阅读量206

点赞数

CC 4.0 BY-SA版权

分类专栏： # OJ 文章标签： c++ 队列算法

本文链接：https://blog.csdn.net/qq_33373173/article/details/123356197

OJ 专栏收录该内容

4 篇文章

订阅专栏

在这里插入图片描述

#include <iostream>
#include <cstdio>
#include <vector> //边的信息使用邻接表
#include <queue> //拓扑排序使用队列
using namespace std;

int n; //点的数量
int m; //边的数量
int inDegree[100001]; //保存入度数量
int ans[100001]; //最终的答案

struct edge { //边的信息
    int e; //边终点
    int v; //边的权值
};

int main()
{
    scanf("%d%d", &n, &m);
    vector<vector<int> > edg(n + 1, vector<int>()); //邻接表存储边的信息
    for (int i = 0; i < m; ++i) {
        int a, b;
        scanf("%d%d", &a, &b);
        ++inDegree[b];
        edg[a].emplace_back(b);
    }

    queue<int> que;
    for (int i = 1; i <= n; ++i) 
        if (0 == inDegree[i]) //起点
            ans[i] = 1, que.push(i);

    while (!que.empty()) {
        int tmp = que.front();
        que.pop();
        for (int i = 0; i < edg[tmp].size(); ++i) { //邻接表下，遍历这个点为起点的所有边
            int e = edg[tmp][i];
            ans[e] = ans[tmp] + 1;
            --inDegree[e];
            if (0 == inDegree[e]) que.push(e);
        }
    }
    for (int i = 1; i <= n; ++i)
        printf("%d\n", ans[i]);
    return 0;
}