프로그래머스 [lv2] 거리두기 확인하기 파이썬

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

 

📗  접근

우선 세가지 함수로 나누어 접근했습니다.

 

1) 각 matrix를 순회

2) 순회를 하면서 대기실의 범위 내부만 탐색하도록

3) 거리두기를 지켰는지   -> main 이 되는 알고리즘

 

 

코드

1) 각 matrix를 순회: solution(places)

def solution(places):
    answer = []
    entire_matrix = [(i, j) for i in range(5) for j in range(5)]
    for place in places:
        for r, c in entire_matrix:
            if place[r][c] == 'P' and not check(r, c, place):
                answer.append(0)
                break
        else:
            answer.append(1)

    return answer

대기실의 각 위치를 좌표로 갖는 배열을 생성합니다 : entire_matrix

# entire_matrix
[(0, 0), (0, 1), (0, 2), (0, 3), (0, 4), (1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (2, 0), (2, 1), (2, 2), (2, 3), (2, 4), (3, 0), (3, 1), (3, 2), (3, 3), (3, 4), (4, 0), (4, 1), (4, 2), (4, 3), (4, 4)]

그리고 for 문을 통해 차례대로 places 를 순회합니다.

entire_matrix의 원소를 좌표로 사용하여, P가 존재하면서, 거리두기를 지키는지 여부를 확인합니다.

거리두기의 여부를 확인하는 함수는 check()로 명명하였습니다.

for-else 구문을 사용하여, for 문을 모두 통과하면 거리두기를 제대로 지켰다는 뜻이므로 else 구문에서 append(1)을 합니다.

 

 

2) 대기실의 범위 내부만 탐색: in_range()

def in_range(x, y):
    SIZE = 5
    return -1 < x < SIZE and -1 < y < SIZE

 

3) 거리두기 확인: check()

케이스 별로 나누어 생각해야합니다.

1️⃣ P 를 기준으로, 동서남북 한칸씩의 범위 ➡️ 이 범위에 또 다른 P 가 있다면 무조건 False

2️⃣ P 를 기준으로, 대각선으로 한칸 범위 

3️⃣ P 를 기준으로, 동서남북 두칸씩의 범위 ➡️ 두 칸 떨어진 곳에 P 가 있을 때, 기준점과의 사이에 "X"가 없다면 False

 

대충 이런 그림

 

def check(r, c, p):
    #1
    dist = [(1, 0), (0, 1), (-1, 0), (0, -1)]
    for dx, dy in dist:
        nx, ny = r + dx, c + dy
        if in_range(nx, ny) and p[nx][ny] == 'P':
            return False
            
    #2    
    dist = [(-1, -1), (-1, 1), (1, -1), (1, 1)]
    for dx, dy in dist:
        nx, ny = r + dx, c + dy
        if in_range(nx, ny) and p[nx][ny] == "P" and (p[r][ny] != "X" or p[nx][c] != "X"):
            return False
            
    #3
    dist = [(2, 0), (0, 2), (-2, 0), (0, -2)]
    for dx, dy in dist:
        nx, ny = r + dx, c + dy
        if in_range(nx, ny) and p[nx][ny] == "P" and p[r + dx // 2][c + dy // 2] != "X":
            return False
        
    return True

모든 케이스를 다 통과하면 마지막에 True 를 리턴합니다. 이는 인근 범위 내에서 거리두기가 잘 지켜졌음을 의미합니다.

 

🍎  전체 코드

def in_range(x, y):
    SIZE = 5
    return -1 < x < SIZE and -1 < y < SIZE


def check(r, c, p):
    dist = [(1, 0), (0, 1), (-1, 0), (0, -1)]
    for dx, dy in dist:
        nx, ny = r + dx, c + dy
        if in_range(nx, ny) and p[nx][ny] == 'P':
            return False
        
    dist = [(-1, -1), (-1, 1), (1, -1), (1, 1)]
    for dx, dy in dist:
        nx, ny = r + dx, c + dy
        if in_range(nx, ny) and p[nx][ny] == "P" and (p[r][ny] != "X" or p[nx][c] != "X"):
            return False

    dist = [(2, 0), (0, 2), (-2, 0), (0, -2)]
    for dx, dy in dist:
        nx, ny = r + dx, c + dy
        if in_range(nx, ny) and p[nx][ny] == "P" and p[r + dx // 2][c + dy // 2] != "X":
            return False
        
    return True


def solution(places):
    answer = []
    entire_matrix = [(i, j) for i in range(5) for j in range(5)]
    for place in places:
        for r, c in entire_matrix:
            if place[r][c] == 'P' and not check(r, c, place):
                answer.append(0)
                break
        else:
            answer.append(1)

    return answer