[프로그래머스 뿌셔뿌셔] :Kotlin: 크레인 인형 뽑기 게임

🧱 프로그래머스 코틀린 공부 

🛸 크레인 인형 뽑기 게임 

 

문제 : ( 보러가기 ) 

해설 : 

이 문제는 크레인이 다차원 배열로 된 배열의 숫자를 뽑아서 같은 수일 경우 없애준다. 이 없애준 수의 갯수를 리턴하라는 문제이다. 

위의 사진 처럼 인형이 배열되어있다. 이것을 간단히 예로 표현하면 아래와 같다. 

0	0	0	0	0
0	0	1	0	5
0	3	2	0	1
4	3	4	4	3
5	2	1	5	1

이렇게 5*5 형태로 2차원 배열을 준다. 나는 이 부분에서 혼동이 와서 헤맸다.. 주어진 테스트 케이스는 아래와 같이 준다. 

[[00000],[00105],[03201],[43443],[52151]] 이부분을 반대로 생각해서 많이 해맸지만 이 부분을 어렵게 보지말고 위의 표처럼 쭈욱 쌓여있다는 상태로 본다면 이해가 빠르다. 

 

이 배열 말고도 크레인의 위치를 나타내 주는 배열이 하나 더 주어진다. 

[1, 5, 1, 3, 2, 1, 2] 라고 예를 들면, 

크레인은 1번 위치인 4를 가져온다. 무슨말이냐 하면 0 0 0 0 0 이 다차원 배열의 0번째 인자이지만 여기서 맨 뒤의 것을 가져오는 것이 아니라, 위의 표 상에서의 가장 위의 값인 4를 가져온다고 보면된다. 즉 다차원 배열 상에서는 

3번째 인자에서 0번째 요소인 4를 가져오는 결과가 된다. 

(혹시 이해가 안된다면 댓글.) 

 

이제 이해가 되니 다시 복기를 해보겠다. 

크레인의 좌표와 다차원 배열의 좌표를 정상 가동되게 코딩을 해줘야 한다. 

예를 들어 크레인 위치가 1이라면, 다차원배열의 모든~ 0번째 인자들 중 0이 아닌 값에서 처음 나오는 값이 스택에 쌓여야 한다. 

 

두 가지 개념을 이해해야 한다. 

• 스택 
• 다차원 배열에서의 이중 반복문

스택은 책을 쌓는 듯이 데이터를 넣어준다. 그리고 pop, push, peek 등의 메서드를 통해서 값을 꺼냈다 넣었다 지웠다 할 수 있다. 그리고 이러한 기준 점은 제일 마지막 값이 된다. FILO 의 개념을 알면 간단하다. 

다차원 배열의 경우 보통 [][] 이렇게 두가지 주소값을 넣을 공간이 생긴다. 앞의 [] 는 첫 요소의 위치, 뒤의 [] 는 해당 요소 안에 요소의 위치라고 알 수 있다. 

이를 어떻게 활용하느냐에 따라 반대로 움직일 수도 있다. 

 

이제 코드를 보자. ( 참고로 이 풀이는 나도 답안을 추출한 것이다. 다른 분의! ) 

import java.util.* 
class Solution{
    fun solution(board: Array<IntArray>, moves: IntArray): Int{
        var answer = 0 
        val stack = Stack<Int>() 
        moves.forEach {  
            for(i in board.indices) { 
                if(board[i][it-1] != 0 ){                     
                    if(stack.isNotEmpty() && stack.peek() == board[i][it-1]){                        
                        answer += 2
                        stack.pop()
                    }else{
                        stack.push(board[i][it-1])
                    }
                    board[i][it-1] = 0 
                    break
                }            
            }
         }
        return answer
    }
}

먼저, java.util.* 을 import해준다. 

이제, stack을 선언해준다.  val stack = Stack<Int>()

 

크레인의 움직임인  moves를 기준으로 돌리기 위해 가장 먼저 반복문을 돌려준다. moves.forEach{}

 

다음은 인형들이 들어있는 보드를 인덱스 기준으로 돌려준다. for(i in board.indices) {}

이렇게 하면 크레인의 위치를 기준으로 보드판안을 돌아다닌다고 보면된다. 

 

이제 조건문을 통해서 원하는 조건으로 튜닝해준다. 

if(board[i][it-1] != 0)  이 조건문은 0의 값을 걸러준다. 그리고 i가 기준이다. i는 보드의 인덱스이기 때문에

모든 요소들의 0번째 의 인자를 먼저 쭈욱 돌 것이다. 그리고 여기서 크레인의 위치가 +1 상태이기 때문에 -1을 해주어 it -1 을 해준다. 이렇게 하면 위의 표에서 세로 방향으로 0이 아닌 값 첫번째를 찾게 된다. 

그래서 처음 stack에 쌓이게 되는 값은 4가 된다. stack.push(board[i][it-1]) 

 

위의 코드를 보면 이 else 부분이 먼저 적용된다. 

그러다가 stack에 같은 값이 쌓이게 되면 삭제하고 삭제된 요소의 갯수를 세어주는 조건을 달아준다. 

if(stack.isNotEmpty() && stack.peek() == board[i][it-1])  

스택이 비어있지 않거나, 스택에서 맨 뒤 값을 가져오고, 그 값이 해당 0이 아닌 값과 같다면 

answer+2 를 해준다. 

그리고 해당 값은 지워주기 위해 stack.pop() 을 해준다. 

 

자 이렇게 되었으면 해당 값은 Stack으로 들어갔기 때문에, board[i][it-1] = 0으로 설정하여 빈 값으로 설정!

 

이제 return answer 를 하면 원하는 값이 나온다.