개발 학습 블로그입니다.

4 minute read

프로그래머스 2021 카카오 채용 연계형 인턴쉽 문제 중 81302번 ‘거리두기 확인하기’

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

문제설명

개발자를 희망하는 죠르디가 카카오에 면접을 보러 왔습니다.

코로나 바이러스 감염 예방을 위해 응시자들은 거리를 둬서 대기를 해야하는데 개발 직군 면접인 만큼 아래와 같은 규칙으로 대기실에 거리를 두고 앉도록 안내하고 있습니다.

  1. 대기실은 5개이며, 각 대기실은 5x5 크기입니다.
  2. 거리두기를 위하여 응시자들 끼리는 맨해튼 거리1가 2 이하로 앉지 말아 주세요.
  3. 단 응시자가 앉아있는 자리 사이가 파티션으로 막혀 있을 경우에는 허용합니다. 예를 들어,

FireShot Capture 002 - 코딩테스트 연습 - 거리두기 확인하기 - 프로그래머스 - programmers.co.kr.png

5개의 대기실을 본 죠르디는 각 대기실에서 응시자들이 거리두기를 잘 기키고 있는지 알고 싶어졌습니다. 자리에 앉아있는 응시자들의 정보와 대기실 구조를 대기실별로 담은 2차원 문자열 배열 places가 매개변수로 주어집니다. 각 대기실별로 거리두기를 지키고 있으면 1을, 한 명이라도 지키지 않고 있으면 0을 배열에 담아 return 하도록 solution 함수를 완성해 주세요.

제한사항

  • places의 행 길이(대기실 개수) = 5
    • places의 각 행은 하나의 대기실 구조를 나타냅니다.
  • places의 열 길이(대기실 세로 길이) = 5
  • places의 원소는 P,O,X로 이루어진 문자열입니다.
    • places 원소의 길이(대기실 가로 길이) = 5
    • P는 응시자가 앉아있는 자리를 의미합니다.
    • O는 빈 테이블을 의미합니다.
    • X는 파티션을 의미합니다.
  • 입력으로 주어지는 5개 대기실의 크기는 모두 5x5 입니다.
  • return 값 형식
    • 1차원 정수 배열에 5개의 원소를 담아서 return 합니다.
    • places에 담겨 있는 5개 대기실의 순서대로, 거리두기 준수 여부를 차례대로 배열에 담습니다.
    • 각 대기실 별로 모든 응시자가 거리두기를 지키고 있으면 1을, 한 명이라도 지키지 않고 있으면 0을 담습니다.

FireShot Capture 003 - 코딩테스트 연습 - 거리두기 확인하기 - 프로그래머스 - programmers.co.kr.png

입출력 예 설명

입출력 예 #1

첫 번째 대기실

No. 0 1 2 3 4
0 P O O O P
1 O X X O X
2 O X X O X
3 O O X O X
4 P O X X P
  • 모든 응시자가 거리두기를 지키고 있습니다.

두 번째 대기실

No. 0 1 2 3 4
0 P X O P X
1 O X O X P
2 O X P O X
3 O X X O P
4 P X P O X
  • (0, 0) 자리의 응시자와 (2, 0) 자리의 응시자가 거리두기를 지키고 있지 않습니다.
  • (1, 2) 자리의 응시자와 (0, 3) 자리의 응시자가 거리두기를 지키고 있지 않습니다.
  • (4, 3) 자리의 응시자와 (4, 4) 자리의 응시자가 거리두기를 지키고 있지 않습니다.

세 번째 대기실

No. 0 1 2 3 4
0 P X O P X
1 O X O X P
2 O X P O X
3 O X X O P
4 P X P O X
  • 모든 응시자가 거리두기를 지키고 있습니다.

네 번째 대기실

No. 0 1 2 3 4
0 O O O X X
1 X O O O X
2 O O O X X
3 O X O O X
4 O O O O O
  • 모든 응시자가 거리두기를 지키고 있습니다.

다섯 번째 대기실

No. 0 1 2 3 4
0 P X P X P
1 X P X P X
2 P X P X P
3 X P X P X
4 P X P X P
  • 모든 응시자가 거리두기를 지키고 있습니다.

풀이

입력으로 주어진 거리두기 각각의 배치도들 중 응시자가 있는 각각의 포인트 P에 대하여 거리가 2인 탐색을 모두 진행하면 알 수 있는 그래프 탐색 문제였다.

코드의 흐름은 다음과 같다.

  1. 입력으로 주어진 places 중 각각의 대기실 배치도에서 응시자가 있는 위치를 찾고 저장할 함수 get_starts를 선언한다.
  2. 지금 조사할 응시자의 위치 s와 대기실의 상태 p를 입력받아 지금 대기실에서 현재 응시자가 거리두기를 잘 지키는지 확인할 함수 is_distance를 선언한다.
  3. 입력으로 주어진 places에 대해 각각의 대기실 place에 대해서 각 place에 존재하는 모든 응시자들인 get_starts(place)의 모든 성분에 대해 True, False를 판별하고 하나라도 False 경우 거리두기가 지켜지지 않았으므로 answer에 0을 저장한다.

내부함수에 대해 조금 더 자세히 설명하면 다음과 같다.

  • get_starts
    1. 입력으로 주어진 p, 5 x 5 배열의 모든 원소에 대해 ‘P’인 지점의 인덱스와, 응시자로 부터 거리를 나타낼 INITIAL_DISTAMCE 을 points에 저장한다.
    2. 배열 points를 return한다.
  • is_distance
    1. 입력으로 주어진 응시자의 시작점 s가 존재하지 않으면 거리두기가 지켜지고 있으므로 바로 True를 리턴하고 조기종료한다.
    2. 방문여부를 체크할 visited 배열을 선언한다.
    3. BFS를 위한 큐를 선언하고, s로 초기화한다.
    4. 응시자의 위치로부터 거리를 나타내는 d가 2보다 작을때만 BFS를 진행하고 이떄, 탐색 중 O를 만나면 거리 d를 최신화하며 큐에 넣어주고, P를 만난다면 거리두기가 지켜지지 않으므로 False를 반환하면 탐색을 종료한다.
from typing import List
from collections import deque


def solution(places: List[List[str]]) -> List[int]:
     # 상수 선언
    ROW, COL = 5, 5
    dr, dc = [1, 0, -1, 0], [0, 1, 0, -1]
    
     # 초기 응시자의 위치를 구할 함수 선언.
    def get_starts(p: List[List[str]]) -> List[List[int]]:
        points = []
         # 응시자로부터의 거리를 0으로 초기화
        INITIAL_DISTAMCE = 0
        for i in range(5):
            for j in range(5):
                if p[i][j] == "P":
                     # 응시자의 좌표와 거리를 points 배열에 추가
                    points.append([i, j, INITIAL_DISTAMCE])
        return points
    
    
    def is_distance(s: List[int], p: List[str]) -> bool:
         # s가 없다면 return True 조기종료
        if not s:
            return True

         # 방문 여부를 확인하기 위한 배열 선언
        visited = [[False] * COL for _ in range(ROW)]
        q = deque([s])
        while q:
            r, c, d = q.popleft()
             # 방문표시
            visited[r][c] = True
             # 거리가 2보다 크다면 탐색을 진행하지 않음
            if d < 2:
                for i in range(4):
                    nr, nc = r + dr[i], c + dc[i]
                    if 0 <= nr < ROW and 0 <= nc < COL and not visited[nr][nc]:
                         # 'P'를 만나면 거리두기 실패, return False
                        if p[nr][nc] == 'P':
                            return False
                         # 'O'를 만난다면 거리 d를 1증가시키고 다음 탐색을 위해 큐에 넣어줌
                        if p[nr][nc] == 'O':
                            q.append([nr, nc, d + 1])
        return True
    
    n = len(places)
    answer = [1] * n
    for idx, place in enumerate(places):
        for start in get_starts(place):
            if not is_distance(start, place):
                answer[idx] = 0
    return answer
            

solution(
    [["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"]]
)

[1, 0, 1, 1, 1]

Leave a comment

You may also enjoy