Lv.2 : 주식가격

👉 문제링크

프로그래머스

---

---

🔸 문제 분석 🔸

• 입력 된 가격 배열에서 각 인덱스의 가격이 몇 초동안 떨어지지 않았는지 반환한다.

🔸 문제 풀이 🔸

• 스택에 각 가격 별 인덱스를 순서대로 저장한다.
• 현재 스택에 저장 된 인덱스의 가격이 현재 인덱스 가격보다 크다면 가격이 떨어진 것이므로, 스택에서 인덱스를 꺼내고 지나간 초를 저장한다.

---

🔸 코드 🔸

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[stack.peek()] > prices[i])
                answer[stack.peek()] = i - stack.pop();

            stack.push(i);
        }

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

        return answer;
    }
}

🔸 코드 해석 🔸

• 주어진 주식 가격 배열 prices의 원소를 하나씩 확인하며 계산한다.
  • 스택의 인덱스 값보다 현재 인덱스의 값이 더 작다면, 스택의 인덱스를 꺼내 경과된 시간을 정답 배열 answer에 저장한다.
  • 현재 인덱스를 스택에 저장한다.
• 모든 반복이 끝났으면, 스택에 남은 값들을 정리해준다.
  • 현재 시간을 prices의 길이로 계산해서 경과된 시간을 저장한다.
• answer을 반환한다.

---

🔸 end 🔸

• 간단한 스택문제였다. 값을 저장할까 인덱스를 저장할까 정도만 고민이 됐던 것 같다.