想要精通算法和SQL的成长之路 - 每日温度

前言

想要精通算法和SQL的成长之路 - 系列导航

一. 每日温度

原题链接

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

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

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

1.1 暴力法

我们先从暴力法的角度来考虑：

1. 我们循环遍历每一天温度，以当前温度为基准。再起一个遍历，循环之后的温度。
2. 找到第一个温度大于基准温度的，第二层遍历就退出循环。两层循环的索引（下标）差就是基准温度想观测到更高的气温，至少需要等待的天数。

代码如下：

public int[] dailyTemperatures(int[] temperatures) {
    int len = temperatures.length;
    int[] res = new int[len];
    // 遍历每一个元素
    for (int i = 0; i < len; i++) {
        int cur = temperatures[i];
        // 从当前元素继续往后遍历，找到第一个比它大的元素即可跳出循环
        for (int j = i + 1; j < len; j++) {
            if (temperatures[j] > cur) {
                // 下标距离就是等待天数
                res[i] = j - i;
                break;
            }
        }
    }
    return res;
}

注意：

1. 因为题目给了温度范围，比100度还要高的不存在，所以本题也可以在判断的时候，加一个限制条件，if(cur<100)。

但是暴力法再数据量非常大的时候会超时，因此往往不可取。

1.2 单调栈

首先我们来看下题目，思考一下几个特性：

1. 我们以任何一天的温度A为基准，往后看。题目的要求是：找到第一个比当前温度A大的温度，然后计算它的天数差。
2. 找到温度低的，我们就不管了。我们只关心温度更高的。也就是说温度需是单调递增排序的。
3. 既然要计算天数差，那么我们在使用单调栈的时候，存储的必定不是温度，而是对应的下标索引。

那代码如下：

public int[] dailyTemperatures(int[] temperatures) {
    int len = temperatures.length;
    int[] res = new int[len];
    // 单调递增栈，存储的是下标
    LinkedList<Integer> stack = new LinkedList<>();
    for (int i = 0; i < len; i++) {
        // 如果当前天气比栈顶下标对应的温度要高，说明当天是栈顶下标所需的第一个更高气温的天数。
        while (!stack.isEmpty() && temperatures[stack.peek()] < temperatures[i]) {
            // 计算对应下标对应的所需天数，最后弹出栈
            res[stack.peek()] = i - stack.peek();
            stack.pop();
        }
        Integer pop = stack.pop();
        // 下标入栈
        stack.push(i);
    }
    return res;
}