[알고리즘 || 프로그래머스] 크레인 인형 뽑기 게임

오늘은 강의에서 배운 대로

먼저 어떤게 주어졌고, 어떤 걸 구해야 하는지 정리를 먼저 해두고, 

그 다음에 어떤 방식으로 구해야 하는지 써봤다. 

이 과정도 테스트의 일부라고 생각하고 쓰니 단계별로 차근차근 쓸 수 있었다. 

 

0. 2차원 배열과 1차원 배열이 주어진다. / return 하는 건 count 된 숫자
1. 2차원 배열에서 moves 배열의 행 중 가장 위에 있는 수 중 0 이 아닌 수 를 찾는다.
2. 임의의 바구니 array 를 만든다.
3. 1번에서 찾은 0이 아닌 수를 array 에 넣는다.
4. 바구니에서 같은 수2개가 있다면 제거한다.
5. 제거 된 인형의 개수를 구한다. (제거되는 과정이 반복된 수 *2 를 해도 됨)

 

4번은 따로 진행하는 것보다 3번을 진행하면서 확인하며 진행하는게 더 간편할거란 생각이 들었다.

function solution(board, moves) {
    const basket = [];
    let count = 0;
    moves.forEach(move => {
        for (let i = 0; i < board.length; i++) {
            if (board[i][move - 1] !== 0) {
                if (basket[basket.length - 1] === board[i][move - 1]) {
                    count++;
                    basket.pop();
                } else {
                    basket.push(board[i][move - 1]);
                }
                board[i][move - 1] = 0;
                break;
            }
        }
    });
    return count * 2
}

 

moves 의 수들을 기준으로 board 에서 수를 찾는다.

moves 의 수인 move 를 고정해놓은 상태에서

for 문을 돌려 2차원 배열인 board 를 탐색한다.

0이 아닌 수일 경우

- 만약 basket 에 있는 마지막 수와 발견한 수가 같다면, basket 에서 그 수를 제거해주고 count 를 +1 해준다.

- 만약 basket 에 있는 마지막 수와 발견한 수가 다르다면, basket 에 수를 추가해준다.

어쨌든 0이 아닌 수가 나왔기 때문에 뽑기가 실행되고, board 의 자리에 0을 넣어준 뒤, break 로 for 문을 중단한다.

 

count 는 같은 수를 만나 제거가 실행된 수이기 때문에, 

인형의 수를 return 하기 위해 *2를 곱한다.

인형은 2개가 만나면 무조건 사라지기 때문에 3콤보 등은 고려할 필요가 없다.