[Programmers] Level 2. 지게차와 크레인

🔗 Problem Link

프로그래머스

---

❔Thinking

• 컨테이너를 빼내는 요청이 주어질 때, 모든 요청을 수행하고 남은 컨테이너 수를 반환한다.
• 컨테이너는 빼내는 요청은 두가지로 나뉜다.
  • 알파벳 1개 : 해당 컨테이너를 지게차로 꺼낸다. 지게차는 외부와 맞닿은 컨테이너만 꺼낼 수 있다.
  • 알파벳 2개 : 해당 컨테이너를 크레인으로 꺼낸다. 크레인은 모든 컨테이너를 조건없이 꺼낼 수 있다.

---

💻Solution

from collections import deque
def solution(storage, requests):
    
    def fork_lift(target:str):
        nonlocal left_container
        tmp_container = []
        visited = [[False] * (M+2) for _ in range(N+2)]
        visited[0][0] = True
        q = deque([(0,0)])
        while q:
            r, c = q.popleft()
            for nr, nc in [(r-1,c),(r+1,c),(r,c-1),(r,c+1)]:
                if 0<=nr<=N+1 and 0<=nc<=M+1 and visited[nr][nc] is False:
                    if storage[nr][nc] == target:
                        left_container -= 1
                        visited[nr][nc] = True
                        tmp_container.append((nr,nc))
                    elif storage[nr][nc] == '#':
                        q.append((nr,nc))
                        visited[nr][nc] = True
        for i,j in tmp_container:
            storage[i][j] = '#'
        return

    def crain(target:str):
        nonlocal left_container
        cnt = 0
        tmp_container = []
        for i in range(1,N+1):
            for j in range(1,M+1):
                if storage[i][j] == target:
                    tmp_container.append((i,j))
                    left_container -= 1
        for i, j in tmp_container:
            storage[i][j] = '#'
        return
    
    N, M = len(storage), len(storage[0])
    left_container = N*M
    storage = [['#']*(M+2)] + [['#']+list(s)+['#'] for s in storage] + [['#']*(M+2)]
    
    for request in requests:
        if len(request) == 1:
            fork_lift(request[0])
        elif len(request) == 2:
            crain(request[0])
            
    return left_container

---

🗝️keypoint

1. 지게차로 컨테이너를 빼내는 방법에 유의한다. 외부와 맞닿은 컨테이너인지 확인하는 방법은, BFS를 통해 외부에서 접근 가능하다면 꺼내는 방법이다.
2. 2차원 이상의 배열에서 조건에 맞는 위치의 값을 변경할때는, 전부 확인 후 값을 변경할지, 해당 조건에 맞으면 곧바로 값을 변경할지 선택해야 한다.