[Programmers] Lv1. 크레인 인형 뽑기(kotlin)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/64061?language=kotlin

 

 

예시

https://school.programmers.co.kr/learn/courses/30/lessons/64061?language=kotlin

 

 

풀이

• 전형적인 스택을 이용하여 풀 수 있는 문제입니다.
• 스택 리스트를 하나 생성한 후 뽑는 순서에 맞게 스택에서 하나 제거시킵니다.
• 제거된 숫자를 다른 스택에 저장해두고 만약 제거된 숫자와 스택의 최상단 숫자가 같을 경우 제거하고 answer에 2를 더합니다.

import kotlin.math.*
import java.util.*

class Solution {
    fun solution(board: Array<IntArray>, moves: IntArray): Int {
        var answer = 0
        
        val stackList = List(board.size){ Stack<Int>() }
        board.reverse()
        
        for(i in 0 until board.size){
            for(j in 0 until board.size){
                if(board[i][j] != 0){
                    stackList.get(j).add(board[i][j])
                }
            }
        }
        
        val s = Stack<Int>()
        
        moves.forEach{
            if(stackList.get(it - 1).isNotEmpty()){
                if(s.isEmpty()){
                    s.add(stackList.get(it-1).pop())
                }else{
                    if(s.peek() == stackList.get(it-1).peek()){
                        answer+=2
                        stackList.get(it-1).pop()
                        s.pop()
                    }else{
                        s.add(stackList.get(it-1).pop())
                    }
                }
            }
        }
        
        return answer
    }
}

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• 참고로 kotlin에서는 Stack 자료구조는 없어서 java util 라이브러리에서 가져와서 사용하거나 리스트를 스택과 유사하게 만들어서 사용하면 됩니다.
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/-list/

List - Kotlin Programming Language