[구현] 거리두기 확인하기

코딩테스트 연습 - 거리두기 확인하기

def solution(places):
    answer = [1, 1, 1, 1, 1]
    def check_range(x, y):
        # 상하좌우 탐색(1칸)
        dx_1 = [-1, 1, 0, 0]
        dy_1 = [0, 0, -1, 1]
        for i in range(4):
            nx = x + dx_1[i]
            ny = y + dy_1[i]
            if 0 <= nx < 5 and 0 <= ny < 5:
                if graph[nx][ny] == 'P':
                    return 0
 
        # 상하좌우 탐색(2칸)
        dx_3 = [-2, 2, 0, 0]
        dy_3 = [0, 0, -2, 2]
        for i in range(4):
            nx = x + dx_3[i]
            ny = y + dy_3[i]
 
            px = x + dx_1[i]
            py = y + dy_1[i]
            if 0 <= nx < 5 and 0 <= ny < 5:
                if graph[nx][ny] == 'P':
                    if graph[px][py] != 'X':
                        return 0
 
 
        # 대각선 탐색 (파티션 2개)
        target = [[-1, -1], [-1, 1], [1, -1], [1, 1]]
        dx_2 = [[-1, 0], [-1, 0], [1, 0], [1, 0]]
        dy_2 = [[0, -1], [0, 1], [0, -1], [0, 1]]
 
        for i in range(4):
            tx = target[i][0] + x
            ty = target[i][1] + y
            nx1 = x + dx_2[i][0]
            ny1 = y + dy_2[i][0]
            nx2 = x + dx_2[i][1]
            ny2 = y + dy_2[i][1]
 
            if 0 <= tx < 5 and 0 <= ty < 5:
                if graph[tx][ty] == 'P':
                    if 0 <= nx1 < 5 and 0 <= nx2 < 5 and 0 <= ny1 < 5 and 0 <= ny2 < 5:
                        if graph[nx1][ny1] != 'X' or graph[nx2][ny2] != 'X':
                            return 0
        return 1
    for index, v in enumerate(places):
        graph = [list(map(str, i)) for i in v]
 
        for i in range(5):
            for j in range(5):
                if graph[i][j] == 'P':
                    if check_range(i, j) == 0:
                        answer[index] = 0
                        break
    return answer
 
 
a = [["POOOP", "OXXOX", "OPXPX", "OOXOX", "POXXP"], ["POOPX", "OXPXP", "PXXXO", "OXXXO", "OOOPP"],
     ["PXOPX", "OXOXP", "OXPOX", "OXXOP", "PXPOX"], ["OOOXX", "XOOOX", "OOOXX", "OXOOX", "OOOOO"],
     ["PXPXP", "XPXPX", "PXPXP", "XPXPX", "PXPXP"]]
print(solution(a))

---