[Lv. 2] 뒤에 있는 큰 수 찾기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/154539

프로그래머스

난이도 : Level.2

언어 : Kotlin

 

문제 설명 :

정수로 이루어진 배열 numbers가 있습니다. 배열 의 각 원소들에 대해 자신보다 뒤에 있는 숫자 중에서 자신보다 크면서 가장 가까이 있는 수를 뒷 큰수라고 합니다.

정수 배열 numbers가 매개변수로 주어질 때, 모든 원소에 대한 뒷 큰수들을 차례로 담은 배열을 return 하도록 solution 함수를 완성해주세요. 단, 뒷 큰수가 존재하지 않는 원소는 -1을 담습니다.

 

제한 사항 : 

• 4 ≤ numbers의 길이 ≤ 1,000,000
  • 1 ≤ numbers[i] ≤ 1,000,000

입출력 예 :

numbers	result
[2, 3, 3, 5]	[3, 5, 5, -1]
[9, 1, 5, 3, 6, 2]	[-1, 5, 6, 6, -1, -1]

다른 사람의 풀이 : 

import java.util.Stack

class Solution {
    fun solution(numbers: IntArray): IntArray {
        val answer = IntArray(numbers.size) { -1 }
        val stack = Stack<Int>()

        var index = 0
        numbers.forEachIndexed { i, it ->
            while (index <= i) {
                val topNum = if (stack.isNotEmpty()) stack[stack.lastIndex] else -1

                when {
                    stack.isEmpty() || it <= numbers[topNum] -> stack.add(index++)
                    else -> {
                        answer[topNum] = it
                        stack.removeLast()
                    }
                }
            }
        }

        return answer
    }
}

 

풀다가 안풀려서 다른 사람의 풀이를 보았다. 근본적으로 내 풀이와 달랐던 점은 비교의 접근 방식이었던 것 같다.

 

자세한 풀이법 : https://velog.io/@ddanglehee/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%EB%92%A4%EC%97%90-%EC%9E%88%EB%8A%94-%ED%81%B0-%EC%88%98-%EC%B0%BE%EA%B8%B0-in-Kotlin

[프로그래머스] 뒤에 있는 큰 수 찾기 in Kotlin