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

 

📚 문제 해석

n개의 노드가 있고, n개의 줄에 걸쳐 인접 행렬이 주어질때, i에서 j로 가는 간선이 존재하면 1, i에서 j로 가는 간선이 없다면 0을 인접 행렬의 형태로 다시 출력한다.

여기서 인접 행렬이란!! 그래프의 연결 관계를 2차원 배열로 표현하는 방식이다. 간단하게 설명하자면 행은 현재 노드를 의미하고 열은 가고자 하는 노드를 의미한고, 값은 간선을 의미한다. 이 문제에서 이 값이 1이면 둘은 연결관계인것이고, 0이면 연결관계가 없는 것이다. 여기서 연결관계의 의미는 i에서 j로 갈 수 있다는 것이다.

더 궁금하면 요 글을 한번 읽어보시길ㅎ.ㅎ

 

💻 코드

import sys
input = sys.stdin.readline

# 재귀
def dfs(num):
    for i in graph[num]:
        if visited[i] == 0: # 방문한 적 없다면 방문
            visited[i] = 1 # 방문 체크
            dfs(i)

n = int(input())
graph = [[] for i in range(n)]
for i in range(n):
    nums = list(map(int,input().rstrip().split()))
    for j in range(n):
        if nums[j] == 1: # i에서 j로 가는 간선이 존재한다면
            graph[i].append(j) # 그래프에 추가
            
for i in range(n):
    visited = [0 for i in range(n)] # 방문 체크
    dfs(i) 
    print(*visited)

 

🔗 문제링크 경로찾기

11403번: 경로 찾기