Posts 크레인 인형뽑기 [Python]
Post
Cancel

크레인 인형뽑기 [Python]

URL https://programmers.co.kr/learn/courses/30/lessons/64061

문제 설명

  • 게임개발자인 “죠르디”는 크레인 인형뽑기 기계를 모바일 게임으로 만들려고 합니다.
  • “죠르디”는 게임의 재미를 높이기 위해 화면 구성과 규칙을 다음과 같이 게임 로직에 반영하려고 합니다.
  • 게임 화면은 “1 x 1” 크기의 칸들로 이루어진 “N x N” 크기의 정사각 격자이며 위쪽에는 크레인이 있고 오른쪽에는 바구니가 있습니다. (위 그림은 “5 x 5” 크기의 예시입니다). 각 격자 칸에는 다양한 인형이 들어 있으며 인형이 없는 칸은 빈칸입니다. 모든 인형은 “1 x 1” 크기의 격자 한 칸을 차지하며 격자의 가장 아래 칸부터 차곡차곡 쌓여 있습니다. 게임 사용자는 크레인을 좌우로 움직여서 멈춘 위치에서 가장 위에 있는 인형을 집어 올릴 수 있습니다. 집어 올린 인형은 바구니에 쌓이게 되는 데, 이때 바구니의 가장 아래 칸부터 인형이 순서대로 쌓이게 됩니다. 다음 그림은 [1번, 5번, 3번] 위치에서 순서대로 인형을 집어 올려 바구니에 담은 모습입니다.
  • 만약 같은 모양의 인형 두 개가 바구니에 연속해서 쌓이게 되면 두 인형은 터뜨려지면서 바구니에서 사라지게 됩니다. 위 상태에서 이어서 [5번] 위치에서 인형을 집어 바구니에 쌓으면 같은 모양 인형 두 개가 없어집니다.
  • 크레인 작동 시 인형이 집어지지 않는 경우는 없으나 만약 인형이 없는 곳에서 크레인을 작동시키는 경우에는 아무런 일도 일어나지 않습니다. 또한 바구니는 모든 인형이 들어갈 수 있을 만큼 충분히 크다고 가정합니다. (그림에서는 화면표시 제약으로 5칸만으로 표현하였음)
  • 게임 화면의 격자의 상태가 담긴 2차원 배열 board와 인형을 집기 위해 크레인을 작동시킨 위치가 담긴 배열 moves가 매개변수로 주어질 때, 크레인을 모두 작동시킨 후 터트려져 사라진 인형의 개수를 return 하도록 solution 함수를 완성해주세요.

제한사항

  • board 배열은 2차원 배열로 크기는 “5 x 5” 이상 “30 x 30” 이하입니다.
  • board의 각 칸에는 0 이상 100 이하인 정수가 담겨있습니다.
    • 0은 빈 칸을 나타냅니다.
    • 1 ~ 100의 각 숫자는 각기 다른 인형의 모양을 의미하며 같은 숫자는 같은 모양의 인형을 나타냅니다.
  • moves 배열의 크기는 1 이상 1,000 이하입니다.
  • moves 배열 각 원소들의 값은 1 이상이며 board 배열의 가로 크기 이하인 자연수입니다.

문제풀이

  • 아래는 입출력예의 board를 표현한것이다.
    1
    2
    3
    4
    5
    
    [0, 0, 0, 0, 0]
    [0, 0, 1, 0, 3]
    [0, 2, 5, 0, 1]
    [4, 2, 4, 4, 2]
    [3, 5, 1, 3, 1]
    
  • 주어지는 board가 열이아니라 행으로 생각해서 문제풀이에 오해가 있었다.
  • 일단, board를 프린트하면 위의 행렬이 나오고, 이는 초기의 인형이 있는 게임화면과 같다.
  • 여기서 moves에서 0이 아닌애를 가져와서 bucket에 넣으면 된다.
  • 처음 moves는 1이고, 1번째 열은 0,0,0,4,3 이므로, 처음 뽑는 인형은 4가 된다.
  • 그럼 뽑고난 4는 0으로(인형이 뽑혔으니, 빈공간으로 변경)해주고 bucket에 넣는다.
  • 뽑은 인형이 bucket이 비어있거나, 마지막 인형이 지금 뽑은 인형과 다르다면 bucket에 넣고, 같다면 bucket의 마지막을 지워주고 지금 뽑은 인형은 bucket에 넣지 않으며, answer를 +2 해준다
  • 마지막에 answer를 리턴해주면 해결
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
def solution(board, moves):
    answer = 0 # 지워진 인형의 갯수 저장
    bucket = [] # 뽑은인형(temp)를 저장할 bucket
    
    for move in moves: # moves의 위치만큼 for문
        
        # board[뽑는 위치(열)][뽑는 위치(행)]
        for i in range(len(board)): # board를 index(행)로 접근,  board[0][0]은 0, board[1][0]도 0 board[2][0]도 0 board[3][0]은 4로 인형은 4를 뽑는다.
            # move가 1부터 시작하므로 1을 일단 빼줌, board[0][0]이 0이면 다음 board[1][0]으로 가게되며 0이 아닌 값을 만나면 인형을 뽑고 (temp) 뽑힌 위치는 0으로 변경
            if board[i][move-1] != 0: 
                temp = board[i][move-1]
                board[i][move-1] = 0
                
                # bucket이 비어있거나 bucket의 마지막이 지금 뽑은 인형과 같지 않으면 bucket에 지금 뽑은 인형이 저장됨(append), break를 안하면 아래의 다른 인형을 또 뽑게됨
                if len(bucket) == 0 or bucket[-1] != temp:
                    bucket.append(temp)
                    break
                # 위의 조건이 맞지않다면 (bucket의 마지막 인형과 지금 뽑은 인형이 같다면) bucket의 마지막 인형은 지워지고 answer는 +2가 됨, break를 안하면  아래의 다른 인형을 또 뽑게 됨
                else :
                    del bucket[-1]
                    answer += 2
                    break
    return answer
1
board
1
2
3
4
5
[[0, 0, 0, 0, 0],
 [0, 0, 1, 0, 3],
 [0, 2, 5, 0, 1],
 [4, 2, 4, 4, 2],
 [3, 5, 1, 3, 1]]
1
2
3
board = [[0,0,0,0,0],[0,0,1,0,3],[0,2,5,0,1],[4,2,4,4,2],[3,5,1,3,1]]
moves = [1,5,3,5,1,2,1,4]
solution(board, moves)
1
4
This post is licensed under CC BY 4.0 by the author.