코틀린-안드로이드

37일차)알고리즘 문제(뒤에 있는 큰수)

songyooho 2024. 6. 30. 23:55

>알고리즘 문제

1.문제

문제 설명

정수로 이루어진 배열 numbers가 있습니다. 배열 의 각 원소들에 대해 자신보다 뒤에 있는 숫자 중에서 자신보다 크면서 가장 가까이 있는 수를 뒷 큰수라고 합니다.
정수 배열 numbers가 매개변수로 주어질 때, 모든 원소에 대한 뒷 큰수들을 차례로 담은 배열을 return 하도록 solution 함수를 완성해주세요. 단, 뒷 큰수가 존재하지 않는 원소는 -1을 담습니다.


제한사항
  • 4 ≤ numbers의 길이 ≤ 1,000,000
    • 1 ≤ numbers[i] ≤ 1,000,000

2)솔루션

class Solution {
    fun solution(numbers: IntArray): IntArray {
        var answer=IntArray(numbers.size){-1}

        for(i in numbers.size-2 downTo 0){
            for(j in i+1..numbers.size-1){
                if(numbers[i]<numbers[j]){
                    answer[i]=numbers[j]
                    break
                }
                //numbers[i]>=numbers[j]인 경우(i<j)
                //ni~nj까지 ni보다 큰 수가 없었는데 aj=-1이면 nj이후에도 ni보다 큰 수가 없다는뜻
                //즉 ai는 -1
                //만약 aj값이 ni보다 크면  ai는 aj
                else{
                    if(answer[j]==-1){
                        answer[i]=-1
                        break
                    }
                    else if(answer[j]>numbers[i]){
                        answer[i]=answer[j]
                        break
                    }
                }
            }
        }
        return answer
    }
}