[프로그래머스/JAVA 자바] 뒤에 있는 큰 수 찾기

 

프로그래머스

 

문제풀이

 

1. 2중 for문을 사용하니 3개의 테스트가 시간초과 발생.

2. stack를 사용하여 비교

3. [프로그래머스/JAVA] 주식가격 (tistory.com) 문제와 유사한 풀이방법

 

 

나의풀이 ( 코드 )

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = new int[numbers.length];
        Stack<Integer> stack = new Stack<>();
        stack.push(0);

        for (int i = 1; i < numbers.length; i++) {
        	
            while (!stack.isEmpty()&&numbers[stack.peek()]<numbers[i]){
                answer[stack.pop()] = numbers[i];
            }
 
            stack.push(i);
        }

        while (!stack.isEmpty()){
            answer[stack.pop()] = -1;
        }
        return answer;
    }
}