[백준 / C++] 경로 찾기 (실버1, 11403)

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

문제

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

입력

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

출력

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

예제 입력 1

3
0 1 0
0 0 1
1 0 0

예제 출력 1

1 1 1
1 1 1
1 1 1

예제 입력 2

7
0 0 0 1 0 0 0
0 0 0 0 0 0 1
0 0 0 0 0 0 0
0 0 0 0 1 1 0
1 0 0 0 0 0 0
0 0 0 0 0 0 1
0 0 1 0 0 0 0

예제 출력 2

1 0 1 1 1 1 1
0 0 1 0 0 0 1
0 0 0 0 0 0 0
1 0 1 1 1 1 1
1 0 1 1 1 1 1
0 0 1 0 0 0 1
0 0 1 0 0 0 0

---

문제 풀이

이번 문제는 dfs, bfs를 사용해 경로를 탐색하는 문제입니다.

 

저는 기본적으로 거리를 구하는 문제에서 bfs를 사용하고, 나머지는 dfs를 사용하기에 이번 문제 또한 dfs로 접근하였습니다.

 

문제에서 원하는 답은 노드별로 방문할 수 있는 노드들을 출력하는 것이기에 visited 배열을 초기화하며 0 ~ N까지 dfs 알고리즘으로 순회하는 방식으로 문제를 해결하게 되었습니다.

 

기본적인 dfs 알고리즘을 구현할 수 있다면 쉽게 풀이 가능한 문제라고 생각합니다.

코드

풀이 시간 : 11m 2s

#include <iostream>
#include <string.h>
#include <vector>
using namespace std;

int n;
vector<vector<int>> adj;
bool visited[102];

void DFS(int u)
{
	for (int v : adj[u])
	{
		if (!visited[v])
		{
			visited[v] = true;
			DFS(v);
		}
	}
}

int main()
{
	cin >> n;

	for (int i = 0; i < n; i++)
	{
		vector<int> list;

		for (int j = 0; j < n; j++)
		{
			int temp;
			cin >> temp;

			if (temp == 1)
			{
				list.push_back(j);
			}
		}

		adj.push_back(list);
	}

	for (int i = 0; i < n; i++)
	{
		DFS(i);

		for (int j = 0; j < n; j++)
		{
			cout << (visited[j] ? 1 : 0) << " ";
		}
		cout << '\n';
		memset(visited, false, sizeof(visited));
	}
}