프로그래머스 :: 연습문제 :: 뒤에 있는 큰 수 찾기

풀이

fun solution(numbers: IntArray): IntArray {

    val result = IntArray(numbers.size){-1}
    val stack = Stack<Pair<Int, Int>>()

    for(numberIndex in numbers.indices){
        val compareNumber = numbers[numberIndex]

        if(stack.isEmpty()){
            stack.push(Pair(numberIndex, compareNumber))
        }
        else{
            /**
             * 설명
             * - stack 이 비어있을때 peek, pop 을 시도하면 EmptyStackException 이 발생한다.
             * - 따라서 while 조건문 내부에서 stack.peek() 를 바로 시도하게 되면 안된다.
             * - 그렇기 떄문에 stack.empty() 를 선행조건으로 두어 먼저 검사하게 처리하여야 한다.
             * -
             * */
            while(stack.isNotEmpty() && stack.peek().second < compareNumber){
                val lastStackedPair = stack.pop()
                result[lastStackedPair.first] = compareNumber
            }
            stack.push(Pair(numberIndex, compareNumber))
        }
    }

    return result
}

---

출처

프로그래머스

 

풀이

GitHub - sggnology/programmers-algorithm: 프로그래머스 알고리즘 코드 정리입니다.