BFS를 활용한, 방향 그래프 최단거리 in C++

Purple·2021년 9월 18일
0

1. 한 정점에서 출발하여, 방향 그래프내에서 각 정점까지 최단거리를 구하는 코드

출발하는 정점은 1번 정점으로 한다.

#include <iostream>
#include <queue>

using namespace std;
int n, m;
int ch[101], dis[101];
vector<int> graph[101];
queue<int> Q;
int main() {
    freopen("input.txt", "rt", stdin);
    cin >> n >> m;
    for(int i=1; i<=m; i++) {
        int a, b;
        cin >> a >> b;
        graph[a].push_back(b);
    }
    Q.push(1);
    ch[1] = 1;
    while(!Q.empty()) {
        int x = Q.front();
        Q.pop();
        for(int i=0; i<graph[x].size(); i++) {
            if(ch[graph[x][i]] == 0) {
                ch[graph[x][i]] = 1;
                Q.push(graph[x][i]);
                dis[graph[x][i]] = dis[x] + 1;
            }
        }
    }
    for(int i=1; i<=n; i++) {
        cout << dis[i] << " ";
    }
    return 0;
}

갈 수 있고(방향 그래프에서 size만큼 for-loop) && 안 가본 곳을 가는 것 자체가(ch가 0인 곳), 그곳의 최소 도달 거리이다.

  • graph[a].push_back(b) : 방향 그래프
  • ch[graph[x][i]] : 해당 위치를 방문하였는지 기록
  • dis[graph[x][i]] : 해당 위치를 방문하였을때의 거리를 기록
profile
안녕하세요.

0개의 댓글