초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요. 제한사항
입출력 예
prices | return |
---|---|
[1, 2, 3, 2, 3] | [4, 3, 1, 1, 0] |
입출력 예 설명
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++) {
// 조건문 1 : stack이 비어있지 않고
// 조건문 2 : 주식 가격이 떨어질 때,
// 즉 '이전 시간대의 가격'이 '이후 시간대의 가격'보다 클 때
while(!stack.empty() && prices[stack.peek()] > prices[i] )
{ int top = stack.pop();
answer[top] = i - top;
//기간 : 현재 인덱스(현재 주식이 감소한 시점) - 주식가격이 떨어지지 않았을 때의 시간
}
stack.push(i);
}
// 이제 stack에는 끝까지 값이 떨어지지 않은 주식의 인덱스들만 존재한다.
// 값을 구하지 못한 인덱스 = 끝까지 값이 떨어지지 않은 주식의 인덱스
while (!stack.empty()){
answer[stack.peek()] = prices.length - stack.pop() -1;
}
return answer;
}
}