leetcode刷题之 栈和队列(5):数组中元素与下一个比它大的元素之间的距离

[LeetCode] Daily Temperatures 日常温度

Given a list of daily temperatures T, return a list such that, for each day in the input, tells you how many days you would have to wait until a warmer temperature. If there is no future day for which this is possible, put 0 instead.

For example, given the list of temperatures T = [73, 74, 75, 71, 69, 72, 76, 73], your output should be [1, 1, 4, 2, 1, 1, 0, 0].

Note: The length of temperatures will be in the range [1, 30000]. Each temperature will be an integer in the range [30, 100].

这道题给了我们一个数组，让我们找下一个比当前数字大的数字的距离，我们研究一下题目中给的例子，发现数组是无序的，所以没法用二分法快速定位下一个大的数字，那么最先考虑的方法就是暴力搜索了，写起来没有什么难度，但是OJ并不答应。实际上这道题应该使用递减栈Descending Stack来做，栈里只有递减元素，思路是这样的，我们遍历数组，如果栈不空，且当前数字大于栈顶元素，那么如果直接入栈的话就不是递减栈了，所以我们取出栈顶元素，那么由于当前数字大于栈顶元素的数字，而且一定是第一个大于栈顶元素的数，那么我们直接求出下标差就是二者的距离了，然后继续看新的栈顶元素，直到当前数字小于等于栈顶元素停止，然后将数字入栈，这样就可以一直保持从下而上递减栈，且每个数字和第一个大于它的数的距离也可以算出来了，参见代码如下：

72  从72判断到 71

69  递减

71 递减

75  75还得等

74 74被75判断

73  73被74判断

class Solution {
    public int[] dailyTemperatures(int[] T) {
        int n = T.length;
        int[] arr = new int[n];
        //建立一个存储角标的栈
        Stack<Integer> stack = new Stack<>();
        for(int i=0; i<n; i++){
            //角标不为空且循环到数组中的值大于栈顶的值
            while(!stack.isEmpty() && T[i] > T[stack.peek()]){
                int index = stack.pop(); 
        //取出栈顶的值即第一个比当前数组值小的值的位置，
        //你要不断循环，因为栈下面的值肯定是从上而下递增的，
        //比它大的值早就被取出来了，你要看能有多少个比当前数组小的值              
                arr[index] = i - index;            
            }
            stack.add(i);
        }
        return arr;
    }
}

//这道题的创新思维在于用栈存储的事数组的角标而不是对应的值

暴力解法

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