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

문제설명

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

제한사항

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

전체코드(Java)

import java.util.*;
import java.util.stream.*;

class Solution {
    public int[] solution(int[] prices) {
        int[] answer = new int[prices.length];
        Stack<Integer> s = new Stack<>();
        // 가장 처음 값을 stack 우선 넣는다.
        s.push(0);
        
        // 비교하는 값보다 큰 값이 stack 상위에 존재한다면, 현재위치에서 해당 위치의 차이를 answer에 저장한다.
        for(int i=1;i<prices.length;i++){
            // stack이 비어있지 않고 스택의 꼭대기 값 인덱스에 위치한 price가 현재 값보다 크다면, 현재위치에서 해당 위치의 차이를 answer에 저장 후 stack에서 제거
            while(!s.isEmpty()&&prices[s.peek()]>prices[i]){
                answer[s.peek()] = i-s.peek();
                s.pop();
            }
            // 비교하는 값의 index를 추가한다.
            s.push(i);
        }
        
        // stack에 있는 index를 순회하며, 주식가격이 떨어지지 않은 기간을 answer에 저장한다.
        for(int i=s.size();i>0;i--){
            answer[s.peek()] = prices.length-s.peek()-1;
            s.pop();
        }
        
        return answer;
    }
}

혹시라도 정정할 내용이나 추가적으로 필요하신 정보가 있다면 댓글 남겨주시면 감사하겠습니다.

오늘도 Jindory 블로그에 방문해주셔서 감사합니다.