백준] 11403 - 경로 찾기

시간 제한 : 1초

메모리 제한 : 256MB

입력

첫째 줄에 정점의 개수 N (1 ≤ N ≤ 100)이 주어진다. 둘째 줄부터 N개 줄에는 그래프의 인접 행렬이 주어진다. i번째 줄의 j번째 숫자가 1인 경우에는 i에서 j로 가는 간선이 존재한다는 뜻이고, 0인 경우는 없다는 뜻이다. i번째 줄의 i번째 숫자는 항상 0이다.

출력

총 N개의 줄에 걸쳐서 문제의 정답을 인접행렬 형식으로 출력한다. 정점 i에서 j로 가는 경로가 있으면 i번째 줄의 j번째 숫자를 1로, 없으면 0으로 출력해야 한다.

소스코드

#include <iostream>
using namespace std;
int graph[100][100], n;
void Floyd()
{
	for (int k = 0; k < n; k++)
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++)
				if (graph[i][k] == 1 && graph[k][j] == 1)
					graph[i][j] = 1;
}
int main(void)
{
	cin >> n;

	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			cin >> graph[i][j];

	Floyd();

	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
			cout << graph[i][j] << ' ';
		cout << '\n';
	}
}

Tip

구현이 쉬운 플로이드-워셜 알고리즘을 이용하면 풀 수 있다. 다만 모든 경로에 대해 최단거리를 구하는 게 아니라, 경로가 연결되었는지 여부만 기록하면 된다.