(Python, Java) 리트코드 - Daily Temperatures

04 Apr 2022 in Algorithm on Leetcode

[문제 링크]

Python 풀이

from typing import List


class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        answer = [0] * len(temperatures)
        stack = []
        for cur_idx, num in enumerate(temperatures):

            while stack and temperatures[stack[-1]] < num:
                idx = stack.pop()
                answer[idx] = cur_idx - idx

            stack.append(cur_idx)

        return answer

Java 풀이

import java.util.Stack;

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int length = temperatures.length;
        int[] answer = new int[length];

        Stack<Integer> stack = new Stack<>();
        for (int curIdx = 0; curIdx < length; curIdx++) {

            while (!stack.isEmpty() && temperatures[stack.peek()] < temperatures[curIdx]) {
                Integer idx = stack.pop();
                answer[idx] = curIdx - idx;
            }
            stack.push(curIdx);
        }

        return answer;
    }

}