[11403] 경로 찾기 (플로이드-워셜)

11403번: 경로 찾기

 

분석

• 최단거리를 구하는 문제가 아니기 때문에 플로이드-워셜 알고리즘에서 업데이트 부분 수정
• S와 E가 모든 중간 경로(K) 중 1개라도 연결되어 있다면 S와 E는 연결 노드로 저장

 

풀이

"""
모든 정점 (i,j) i에서 j로 가는 경로가 있는지 없는지
i에서 j로 가는 경로가 있으면 i번째 줄의 j번째 숫자를 1로, 없으면 0으로 출력
"""
import sys

# 인접행렬 크기
n = int(input())

# 인접행렬
d = array = [[sys.maxsize for j in range(n)] for i in range(n)]

# 인접 행렬에 시작 도시와 종료 도시가 같은 자리에 0 저장
for i in range(n):
    d[i][i] = 0

for i in range(n):
    d[i] = list(map(int, input().split()))
print(d)

for k in range(n): # 경유지 k
    for s in range(n):
        for e in range(n):
            if d[s][k] ==1 and d[k][e]==1:
                d[s][e]=1

for i in range(n):
    for j in range(n):
        if d[i][j] != sys.maxsize:
            print(d[i][j], end=" ")
        else:
            print(0, end=" ")
    print()