[Pogrammers] 주식가격

정확성  테스트
테스트 1 〉	통과 (0.18ms, 52.1MB)
테스트 2 〉	통과 (0.37ms, 51.6MB)
테스트 3 〉	통과 (1.59ms, 52.6MB)
테스트 4 〉	통과 (1.85ms, 53.5MB)
테스트 5 〉	통과 (2.08ms, 53.1MB)
테스트 6 〉	통과 (0.25ms, 52MB)
테스트 7 〉	통과 (1.46ms, 53.2MB)
테스트 8 〉	통과 (1.42ms, 53.2MB)
테스트 9 〉	통과 (0.24ms, 52.1MB)
테스트 10 〉	통과 (1.65ms, 52.6MB)

효율성  테스트
테스트 1 〉	통과 (31.70ms, 77MB)
테스트 2 〉	통과 (26.11ms, 66.7MB)
테스트 3 〉	통과 (34.24ms, 75MB)
테스트 4 〉	통과 (33.67ms, 67.5MB)
테스트 5 〉	통과 (22.44ms, 66MB)

@Test
void 주식가격() {
    Assertions.assertArrayEquals(new int[]{4,3,1,1,0}, 주식가격_solution(new int[]{1,2,3,2,3}));
}

int[] 주식가격_solution(int[] prices) {
    Stack<Integer> stack = new Stack<>();
    int[] result = new int[prices.length];

    int totalTime = prices.length-1;
    for(int i=0; i<=totalTime; i++) {
        while(!stack.isEmpty() && prices[stack.peek()] > prices[i]) {
            int popIdx = stack.pop();
            result[popIdx] = i - popIdx;
        }
        stack.push(i);
    }

    while(!stack.isEmpty()) {
        int popIdx = stack.pop();
        result[popIdx] = totalTime - popIdx;
    }
    return result;
}