알고리즘/문제 풀이

11403번: 경로 찾기

Themion 2022. 1. 5. 16:24

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

 

11403번: 경로 찾기

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

www.acmicpc.net

모든 노드에서 모든 노드로의 경로를 묻는 문제이므로 플로이드-워셜 알고리즘을 사용한다.

#include <cstdio>

#define MAX_N 100

int main() {
    // graph[i][j]: i에서 j로 가는 경로가 있다면 true, 없다면 false
    bool graph[MAX_N][MAX_N];
    // 노드의 개수
    int N;

    // 문제의 조건을 입력받은 뒤
    scanf("%d", &N);
    for (int i = 0; i < N; i++) for (int j = 0; j < N; j++)
        scanf("%d", &graph[i][j]);

    // 플로이드-워셜 실행
    for (int k = 0; k < N; k++)
        for (int i = 0; i < N; i++) for (int j = 0; j < N; j++)
            graph[i][j] |= graph[i][k] && graph[k][j];

    // 모든 정점 i, j에 대해 i에서 j로 가는 경로가 있는지 여부를 출력
    for (int i = 0; i < N; i++) for (int j = 0; j < N; j++)
        printf("%d%c", graph[i][j], (j == N - 1) ? '\n' : ' ');

    return 0;
}