[백준,c++] 14567번 - 선수과목

문제

14567번: 선수과목 (Prerequisite)

 

코드

#include <iostream>
#include <queue>

using namespace std;
int N, M; //N=과목의 수, M=선수 조건의 수
int indegree[500001];    //모든 노드에 대한 진입차수는 0으로 초기화
vector<int>graph[500001];    //각 노드에 연결된 간선 정보를 담기 위한 연결리스트 초기화
int result[500001];    //위상정렬 수행결과를 담을 리스트


void topologySort() {
    queue<int>q;

    for (int i = 1; i <= N; i++) {    //처음에는 진입차수가 0인 노드를 큐에 삽입
        if (indegree[i] == 0) {
            q.push(i);
            result[i] = 1;    //해당 과목 은 바로 들을 수 있음.
        }
    }
    while (!q.empty()) {    //큐가 빌때까지 반복
        int now = q.front();
        q.pop();

        for (int i = 0; i < graph[now].size(); i++) {    //해당 원소와 연결된 노드들의 진입차수에서 1빼기
            indegree[graph[now][i]] -= 1;
            if (indegree[graph[now][i]] == 0) {
                q.push(graph[now][i]);    //새롭게 진입차수가 0이되는 노드를 큐에 삽입
                result[graph[now][i]] = result[now] + 1;
            }
        }
    }

    for (int i = 1; i <= N; i++) {    //위상 정렬을 수행한 결과 출력
        cout << result[i] << ' ';
    }
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin >> N >> M;
    for (int i = 0; i < M; i++) {
        int a, b; cin >> a >> b;
        graph[a].push_back(b);    //정점 A에서 B로 이동가능
        indegree[b] += 1;    //진입 차수를 1 증가
    }
    topologySort();
}

 

풀이

단순 위상 정렬 알고리즘을 수행하면 되는 문제였습니다.