[Medium] 503. Next Greater Element II

문제

https://leetcode.com/problems/next-greater-element-ii/

Next Greater Element II - LeetCode

Given a circular integer array nums (i.e., the next element of nums[nums.length - 1] is nums[0]), return the next greater number for every element in nums.

The next greater number of a number x is the first greater number to its traversing-order next in the array, which means you could search circularly to find its next greater number. If it doesn't exist, return -1 for this number.

 

원형 정수 배열 번호(즉, nums[nums.length - 1]의 다음 요소는 nums[0])가 주어지면 모든 요소에 대해 다음 큰 숫자를 숫자로 반환합니다.
숫자 x의 다음 큰 숫자는 배열에서 다음 순서에 대한 첫 번째 큰 숫자이며, 이는 순환 검색을 통해 다음 큰 숫자를 찾을 수 있음을 의미합니다. 존재하지 않으면 이 번호로 -1을 반환합니다.

 

예시

Example 1:

Input: nums = [1,2,1]
Output: [2,-1,2]
Explanation: The first 1's next greater number is 2; 
The number 2 can't find next greater number. 
The second 1's next greater number needs to search circularly, which is also 2.

Example 2:

Input: nums = [1,2,3,4,3]
Output: [2,3,4,-1,4]

 

제약 조건

Constraints:

• 1 <= nums.length <= 104
• -109 <= nums[i] <= 109

 

해결 과정

for문과 while문으로 모든 요소에 대해 순회를 한다.

idx가 nums.length와 같아지면 idx = 0 으로 원형 순회를 할 수 있도록 한다.

idx가 i와 같아지면 -1을 push하고 while문을 빠져나온다.

현재 요소인 e보다 큰 값이 있다면 큰 값을 push하고 while문을 빠져나온다.

 

해결 코드

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var nextGreaterElements = function(nums) {
    let res = [];
    
    for(let i = 0 ; i < nums.length ; i++){
        let e = nums[i];
        let idx = i + 1;
        
        while(true){
            // 원형 순회
            if(idx === nums.length){
                idx = 0;
            }
            
            if(idx === i){
                res.push(-1);
                break;
            }   
            
            if(e < nums[idx]){
                res.push(nums[idx]);
                break;
            }
            
            idx++;
        }        
    }    
    return res;
};