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

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

프로그래머스

문제 설명

초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.

제한사항

- prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.
- prices의 길이는 2 이상 100,000 이하입니다.

풀이

감소하지 않았다면 스택에 push하고 감소 하는 값이 발생할 경우 pop 해준다

prices 전체를 돌았음에도 스택에 남아 있는 것들은 감소 하지 않은 것들이기 때문에 prices의 길이 - index - 1 을 넣어준다

코드

import java.util.*;
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.pop();
            }
            stack.push(i);
        }
        
         while (!stack.isEmpty()) {
            answer[stack.peek()] = prices.length - stack.pop() - 1;
        }
        return answer;
    }
}