LeetCode第739题：每日温度（Java）

一：题目

根据每日 气温 列表，请重新生成一个列表，对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高，请在该位置用 0 来代替。

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度，都是在 [30, 100] 范围内的整数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/daily-temperatures
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 题目链接：https://leetcode-cn.com/problems/daily-temperatures/

 二：问题描述

此题明显可以用暴力遍历法解，但是当 temperatures 数组很大的时候，则会非常慢，由于最坏时间复杂度是O(n²)，所以不仅力扣过不了，面试官那里也过不了。

 暴力遍历法时间复杂度为O(n²)的主要原因还是计算了太多重复的元素，第一个元素把后面的元素比较一遍，第二个元素又比较一遍，所以可以思考一下能不能把当前元素先暂存起来，如果找到比它大的，就在输出数组中填入两个元素的下标差。

由此我们可以想到栈这个数据结构，把当前元素压入栈，如果有元素大于栈顶元素，则栈顶元素出栈。仅需遍历一遍当前数组就可以求得结果，时间复杂度O(n)。详细步骤在代码中注释。

三：代码实现

public int[] dailyTemperatures(int[] T) {
        if(T.length < 1 || T.length > 30000){
            return new int[0];
        }
        //声明一个栈
        Stack<Integer> st = new Stack<>();
        //返回的数组
        int[] arr = new int[T.length];
        /*因为初始栈是空的，所以可以先把第一个元素添加进去，以便后续比较
            注意：放入栈的是下标，因为如果放数字，就找不到对应在arr数组中的位置
         */
        st.push(0);
        //只用遍历一遍，时间复杂度为 O(N)
        for (int i = 1; i < T.length; i++) {
            //这里要先判断长度是否为空，因为如果为空，peek()方法会抛异常
            //如果当前元素的值大于栈顶的值，则栈顶元素出栈
            while (st.size() != 0 && T[i] > T[st.peek()]){
                /*注意这里要设置中间变量，如果直接用arr[st.pop()] = i - st.pop()，
                则会出栈两次，如果栈里面只有一个元素，则会报EmptyStackException异常
                */
                int temp = st.pop();
                arr[temp] = i - temp;
            }
            //直到栈为空或者当前元素小于栈顶元素，则入栈
            st.push(i);
        }
        //如果较大元素未出栈，则不用管，arr数组对应位置自动为0
        return arr;
    }

四：总结 

此题关键是存放在栈中的元素不能是元素本身，是元素下标，如果是元素本身，则出栈后则没办法在输出数组中记录数据。另外题中很容易出现栈空异常，所以需要对Stack类的方法较熟悉。