[프로그래머스] 주식가격 - Java

https://school.programmers.co.kr/learn/courses/30/lessons/42584

프로그래머스

 

프로그래머스 주식가격 자바 풀이

난이도: Lv2

 

풀이

수를 비교하여 작은 수가 나오면 스택에서 제거 후 몇초 인지 구한다.

스택에 남아있는 경우 작은 수가 없었기 때문에 끝까지 같을 때 몇초인지 구한다.

 

 

코드

import java.util.Stack;
class Solution {
    public int[] solution(int[] prices) {
        int[] answer = new int[prices.length];
        Stack<Integer> stack = new Stack<>();
        for(int i=0;i<prices.length;i++){
            while(!stack.isEmpty()&&prices[i]<prices[stack.peek()]){
                answer[stack.peek()]=i-stack.peek();
                stack.pop();
            }
            stack.push(i);
        }

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