leetcode-patterns-java/LeetCode_Problems/Java/medium/0739. Daily Temperatures.java at main · elchibek5/leetcode-patterns-java · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
// class Solution {
//     public int[] dailyTemperatures(int[] temp) {
//         int n = temp.length;
//         int[] ans = new int[n];

//         Deque<Integer> stack = new ArrayDeque<>();

//         for (int i = 0; i < n; i++) {
//             while (!stack.isEmpty() && temp[i] > temp[stack.peek()]) {
//                 int prev = stack.pop();
//                 ans[prev] = i - prev;
//             }

//             stack.push(i);
//         }

//         return ans;
//     }
// }

class Solution {
    public int[] dailyTemperatures(int[] temp) {
        int n = temp.length;
        int[] nge = new int[n];

        Stack<Integer> st = new Stack<>();
        for (int i = n - 1; i >= 0; i--) {
            while (!st.isEmpty() && temp[st.peek()] <= temp[i]) {
                st.pop();
            }

            if (!st.isEmpty()) {
                nge[i] = st.peek() - i;
            }

            st.push(i);
        }
        return nge;
    }
}