[Algorithm] [Python] BOJ/백준 - 11403 _ 경로 찾기

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으로 출력해야 한다.

 

제출 코드

---

from sys import stdin
from collections import deque

def bfs(start):
    queue = deque([start])
    visited = [0]*n

    while queue:
        node = queue.popleft()
        # 현재 노드와 연결된 노드 확인
        for i in adj_list[node]:
            # 현재 노드와 연결된 노드를 아직 방문하지 않았다면 방문 진행
            if not visited[i]:
                visited[i] = 1
                queue.append(i)

    print(*visited)


n = int(stdin.readline())
adj_list = [[] for _ in range(n)]

# 인접 행렬 -> 인접 리스트
for i in range(n):
    arr = list(map(int, stdin.readline().split())
    for j in range(n):
        if arr[j] == 1:
            adj_list[i].append(j)

for i in range(n):
    bfs(i)

 

문제풀이

---

각각의 노드에 연결된 노드들만 파악하면 되기에 입력받은 인접 행렬을 인접 리스트로 변환하여 접근하였으며 BFS 를 이용하여 문제를 풀었다.

n 개의 노드를 순차적으로 순회하며 해당 노드를 시작점으로 한 BFS 탐색을 진행한다.