剑指Offer || 038.每日温度

题目

请根据每日 气温 列表 temperatures ，重新生成一个列表，要求其对应位置的输出为：要想观测到更高的气温，至少需要等待的天数。如果气温在这之后都不会升高，请在该位置用 0 来代替。

示例 1:

输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

示例 2:

输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]

示例 3:

输入: temperatures = [30,60,90]
输出: [1,1,0]

提示：

• 1 <= temperatures.length <= 105
• 30 <= temperatures[i] <= 100

注意：本题与主站 739 题相同： 力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

LCR 038. 每日温度 - 力扣（LeetCode）

题解

思路一：暴力解法，因为温度是从30-100.使用一个sum数组来存储所有出现的温度，对应的距离当前元素最近的下标。从后向前遍历原始数组。sum[i]，i表示温度，sum[i]表示在原数组中的下标。因为是后序遍历的，因此一定是出现在后面的更高温度。

代码：

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
    	int[] sum = new int[101];
    	int[] ans=new int[temperatures.length];
    	Arrays.fill(sum,Integer.MAX_VALUE);
    	for(int i=temperatures.length-1;i>=0;i--) {
    		int index=Integer.MAX_VALUE;
    		for(int j=temperatures[i]+1;j<101;j++) {
    			if(sum[j]<index) {
    				index=sum[j];
    			}
    		}
    		if(index<Integer.MAX_VALUE) 
    			ans[i]=index-i;//一定要记得-i，因为是从i开始算第多少个温度更高
    		sum[temperatures[i]]=i;
    	}
    	return ans;
    }
}

思路二：单调栈思想，栈中存储下标，栈中数据所代表的温度从栈底到栈顶是从高到低的，从前向后遍历原始数组。栈不空时，当有温度元素大于栈顶元素时，取出栈顶元素，更新ans[栈内index]（代表的是还需要几天才升高）=当前index-栈内index；温度小于栈顶或者栈空时，直接入栈，最后将所有的栈内残留ans[index]=0。

代码：

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int[] ans = new int[temperatures.length];
        Deque<Integer> stack = new ArrayDeque<Integer>();
        for (int i = 0; i < length; i++) {
            int temperature = temperatures[i];
            while (!stack.isEmpty() && temperature > temperatures[stack.peek()]) {
                int prevIndex = stack.pop();
                ans[prevIndex] = i - prevIndex;
            }
            stack.push(i);
        }
        return ans;
    }
}