LeetCode刷题笔记:739.每日温度

1. 问题描述

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指在第 i 天之后,才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。

示例:

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

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

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

2. 解题思路

一开始没读懂题意,现在梳理一下:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]为例:
answer[0] = 1 表示 0 天之后才有更高温度,指的是temperatures[1] = 74
answer[1] = 1 表示 1 天之后才有更高温度,指的是temperatures[2] = 75
answer[2] = 4 表示 4 天之后才有更高温度,指的是temperatures[6] = 76
answer[6] = 0 表示之后温度不再升高
answer[7] = 0 表示之后温度不再升高(最后一天总是0)

思路① 双重循环暴力求解

思路② 单调栈(方向:栈头 ----> 栈底):
栈内顺序:大 ➡ 小
入栈元素与当前栈内栈首元素比较:
若当前元素 大于 栈首元素,则二者下标做差
若当前元素 小于或等于 栈首元素,则入栈

3. 代码实现

双重循环暴力求解:

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

单调栈求解:

class Solution {
	public int[] dailyTemperatures(int[] temperatures) {
    	Deque<Integer> stack = new LinkedList<>();
    	int[] answer = new int[temperatures.length];
    	for (int i = 0; i < temperatures.length; i++) {
    		// 栈不为空 且 入栈元素大于当前栈顶元素
    		while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
    			int preIndex = stack.pop();
    			answer[preIndex] = i - preIndex;
    		}
    		// 放入的是元素下标
    		stack.push(i);
    	}
    	return answer;
    }
}

你可能感兴趣的:(Java,LeetCode,leetcode,算法,java)