Stack2 - 미로

이런 류(bfs)의 문제는 대게 deque를 사용하나 단원이 스택인만큼 리스트를 활용하여 풀었다.

  • from collections import deque를 적어 deque 라이브러리를 사용할 수 있으며, deque를 사용할 경우 리스트 대신 deque()를, pop(0)대신 popleft()를 입력하면 된다.
move=[(-1,0),(1,0),(0,1),(0,-1)]#위 아래 오른 왼

for tc in range(1,int(input())+1):
    N = int(input())
    arr=[list(map(int,input())) for _ in range(N)]
    
    for i in range(N):
        for j in range(N):
            if arr[i][j]==2:
                start,end=i,j

    chk=[]
    chk.append((start,end))
    
    res=0
    while chk:
        x,y=chk.pop(0)
        if arr[x][y]==3:
            res=1
            break  
        arr[x][y]=1
        
        for i in range(4):
            mx=x+move[i][0]
            my=y+move[i][1]
            
            if 0<=mx<N and 0<=my<N and arr[mx][my]!=1:
                chk.append((mx,my))
    
    print(f'#{tc} {res}')

좋은 웹페이지 즐겨찾기