leetcode：#739 每日温度

leetcode：#739 每日温度

题目详情

采用Java实现。

方法一

最容易想到的解题思路：

public class Solution {
    public int[] dailyTemperatures(int[] T) {
        int[] result = new int[T.length];

        for (int i = 0; i < T.length; i++) {
            for (int j = i + 1; j < T.length; j++) {
                if (T[j] > T[i]) {
                    result[i] = j - i;
                    break;
                }
            }
        }

        return result;
    }
}

方法二

维护一个类似于单调递增栈，后入栈的元素总小于等于栈顶元素。
遍历数组T，比对当前元素与栈顶元素的大小：

• 若当前元素 <= 栈顶元素：入栈
• 若当前元素 > 栈顶元素：弹出栈顶元素，两者下标差值即为所求的天数。
  注意递增栈中记录的是 T 的下标。
• 若栈为空，说明前面元素已经求得了需要等待的天数，直接将当前元素入栈即可。

public class Solution {
    public int[] dailyTemperatures(int[] T) {
        int[] result = new int[T.length];
        Stack<Integer> stack = new Stack<>();

        for (int i = 0; i < T.length; i++) {
            while (!stack.isEmpty() && T[stack.peek()] < T[i]) {
                int temp = stack.pop();
                result[temp] = i - temp;
            }
            stack.push(i);
        }

        return result;
    }
}

补充下单调栈的知识：
单调栈：从栈顶到栈底的元素是严格递增（or递减）。

• 单调递增栈：若当前进栈元素为 e，从栈顶开始遍历元素，把小于 e 或者等于 e 的元素弹出栈，直接遇到一个大于 e 的元素或者栈为空为止，然后再把 e 压入栈中。
• 单调递减栈：则每次弹出的是大于 e 或者等于 e 的元素。

单调递增栈可表示入栈元素左边第一个比它大的元素
单调递减栈可表示入栈元素左边第一个比它小的元素

方法三

可以从最后一天向前推到第一天，因为最后一天气温不可能升高，结果直接为0。
再推算倒数第二天的温度，如果比倒数第一天低，那么答案为1，如果比倒数第一天还高，又因为倒数第一天对应的结果为0，所以倒数第二天的结果也应该为0。

• 若T[i] < T[i+1]，那么res[i]=1；
• 若T[i] >= T[i+1]
  • result[i+1]=0，那么result[i]=0;
  • result[i+1]!=0，那就比较T[i]和T[i+1+result[i+1]]（将第i天的温度 与 比第i+1天大的那天的温度进行比较）

public class Solution {
    public int[] dailyTemperatures(int[] T) {
        int[] result = new int[T.length];
        result[T.length - 1] = 0;
        for (int i = T.length - 2; i >= 0; i--) {
            for (int j = i + 1; j < T.length; j += result[j]) {
                if (T[i] < T[j]) {
                    result[i] = j - i;
                    break;
                } else if (result[j] == 0) {
                    result[i] = 0;
                    break;
                }
            }
        }
        return result;
    }
}