백준 11403번 경로 찾기

https://www.acmicpc.net/problem/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으로 출력해야 한다.

 

풀이

기존 BFS와 다른점은 start에서 출발해서 다시 start로 돌아오는 경우가 있는지 없는지를 체크해줘야 한다. 

돌아오는 경우가 없다면 (i,i) 는 0이 되며 경우가 있다면 (i,i)가 1이 되야하기 때문이다.

 

이를 위해 BFS 함수에서 startCycle=False로 주고 후에 pop 한 인자에 start가 있는지 체크하였다.(코드:22줄)

 

코드

#input setting
N=int(input())
graph=[[] for i in range(N+1)]
for i in range(1,N+1):
    j=list(map(int,input().split()))
    for num,k in enumerate(j):
        if k==1:
            graph[i].append(num+1)
 
def bfs(start):
    result=[]
    visited=[False for i in range(N+1)]
    queue=[start]
    startCycle=False
    while queue:
        v=queue.pop(0)
        visited[v]=True
        result.append(v)
        for i in graph[v]:
            if i not in queue and visited[i]==False:
                queue.append(i)
            if startCycle==False and i==start:
                startCycle=True
    if not startCycle:
        del result[result.index(start)]
    return result
 
result=[[0 for i in range(N)] for i in range(N)]
 
#start bfs 1~N
for i in range(1,N+1):
    connect=sorted(bfs(i))
    for j in connect:
        result[i-1][j-1]=1
 
for i in result:
    d=""
    for j in i:
        d+=str(j)+" "
    print(d[:-1])