[python]프로그래머스 - 거리두기 확인하기

[python]프로그래머스 - 거리두기 확인하기

https://school.programmers.co.kr/learn/courses/30/lessons/81302

프로그래머스

 

from collections import deque

def bfs(arr):
    start = [] #'P'인 좌표들 모으기
    for i in range(5):
        for j in range(5):
            if arr[i][j] == 'P':
                start.append([i,j])
                
    #'P'부터 시작해bfs돌리기
    for s in start:
        # print(s)
        q = deque([s])
        visit = [[0]*5 for _ in range(5)]
        distance = [[0]*5 for _ in range(5)]
        visit[s[0]][s[1]] = 1
        # print(visit)
        
        while q:
            ny,nx = q.popleft()
            directy = [-1,1,0,0]
            directx = [0,0,-1,1]
            
            for i in range(4):
                dy = directy[i] + ny
                dx = directx[i] + nx
                
                if 0<=dy<5 and 0<=dx<5 and visit[dy][dx] == 0:
                    if arr[dy][dx] == 'O':
                        q.append([dy,dx])
                        visit[dy][dx] = 1
                        distance[dy][dx] = distance[ny][nx]+1
                        
                    
                    if arr[dy][dx] == 'P' and distance[ny][nx]<=1:
                        return 0
    return 1

#각 케이스마다 bfs를 돌려주자
def solution(places):
    answer = []
    for p in places:
        answer.append(bfs(p))
    return answer

예시 테스트 케이스는 성공했는데, 제출하고 나서 실패한 케이스가 있어서 한참동안 왜 틀렸지..? 했는데

O를 숫자0으로 적어서였다...하하