[알고리즘] 위상정렬

이론정리

문제풀이

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

//위상정렬
vector<int> topologicalSort(int n, vector<int> &indegree, vector<vector<int>> &graph) {
    vector<int> result;
    queue<int> q;

    for (int i = 1; i <= n; i++) {
        if (!indegree[i]) //진입차수가 0이라면
            q.push(i);
    }
    while (!q.empty()) {
        int node = q.front();
        q.pop();

        result.push_back(node); //형재 정점 순서에 삽입

        for (int i = 0; i < graph[node].size(); i++) {
            int next_node = graph[node][i];
            indegree[next_node]--; //연결된 정점의 진입차수를 1씩 감소
            if (!indegree[next_node]) //연결된 정점의 진입차수가 0이 되었다면
                q.push(next_node);
        }
    }
    return result;
}

int main() {
    int n, m, a, b;
    cin >> n >> m;

    vector<int> indegree(n + 1, 0); //각 정점의 진입차수
    vector<vector<int>> graph(n + 1, vector<int>(0)); //인접 리스트 그래프

    while (m--) {
        cin >> a >> b;
        indegree[b]++;
        graph[a].push_back(b);
    }

    vector<int> result = topologicalSort(n, indegree, graph);

    for (int i = 0; i < n; i++)
        cout << result[i] << ' ';
}