본문 바로가기

알고리즘 공부 및 문제 풀이/백준(BOJ)

[boj] 백준 11403 경로 찾기 (c++) - 플로이드 워셜

[문제]

https://www.acmicpc.net/problem/11403

 

11403번: 경로 찾기

가중치 없는 방향 그래프 G가 주어졌을 때, 모든 정점 (i, j)에 대해서, i에서 j로 가는 경로가 있는지 없는지 구하는 프로그램을 작성하시오.

www.acmicpc.net

 

[풀이]

그래프에 있는 모든 정점 (i,j)에 대해 다른 정점으로 가는 경로가 있는지를 구해야 하므로 플로이드-워셜 알고리즘을 이용.

 

[코드]

 

#include <iostream>
#include <algorithm>
#include <map>
#include <string>
#include <queue>
#include <cstring>
#define INF 987654321

using namespace std;

int N;
int adj[101][101];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    cin >> N;
    for(int i=0; i<N; i++){
        for(int j=0; j<N; j++){
            cin >> adj[i][j];
            if(adj[i][j] == 0)
                adj[i][j] = INF;
        }
    }


    //플로이드-워셜
    for (int k=0; k<N; k++){
        for (int i=0; i<N; i++){
            for (int j=0; j<N; j++){
                adj[i][j] = min(adj[i][j], adj[i][k] + adj[k][j]);
            }
        }
    }

    for(int i=0; i<N; i++){
        for(int j=0; j<N; j++){
            if(adj[i][j] == INF)
                cout << "0 ";
            else
                cout << "1 ";
        }
        cout << "\n";
    }

}