2019 카카오 개발자 겨울 인턴십 코딩테스트 - 크레인 인형뽑기 게임

완전탐색을 이용하였고

 

Stack을 사용하여 바구니를 구현하였다.

 

 

언어: Kotlin

 

소스코드:

import java.util.*
 
class Solution {
    fun solution(dollBox: Array<IntArray>, picker: IntArray): Int {
        var answer = 0
        val basket = Basket()
 
        for (pick in picker) {
            for (dolls in dollBox) {
                val pickedDoll = dolls[pick-1]
                if (pickedDoll!=0) {
                    dolls[pick-1] = 0
                    basket.stack(pickedDoll)
                    break
                }
            }
        }
 
        answer = basket.pair*2
 
        return answer
    }
 
    class Basket {
        val things = Stack<Int>()
        var pair = 0
        private set
 
        fun stack(thing: Int) {
            if (things.size>0 && things.lastElement()==thing) {
                things.pop()
                pair++
            }
            else {
                things.push(thing)
            }
        }
 
 
    }
}