[LeetCode/力扣][Java] 0739. 每日温度(Daily Temperatures)

题目描述:

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 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]

思路:从后往前遍历,若后一个元素大于当前元素(T[i+1]>T[i]),则ans[i]=1。若小于,则与后一个元素的上一级比较(T[ans[i+1]+i+1]),直到有大于T[i]的元素出现。特殊情况是相等的时候,要判断相等元素的ans是否为0,若为0则说明该元素的右侧没有更大元素,则ans[i]同样为0;若不为0,则更新距离即可。While循环时注意break条件,不然容易死循环。评论区里另一种用单调栈的方法,我的复现代码版列在了第二种方法中

代码1

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] ans = new int[len];
        ans[len-1] = 0;
        for (int i = len - 2; i >= 0; i--) {
            if (temperatures[i+1] > temperatures[i]) {
                ans[i] = 1;
            } else if (temperatures[i+1] < temperatures[i]){
                int p = i+1+ans[i+1];
                while (p < len) {
                    if (temperatures[p] > temperatures[i]) {
                        ans[i] = p - i;
                        break;
                    } else if (temperatures[p] < temperatures[i]) {
                        if (ans[p] == 0) break;     //这里一定不能漏!
                        else p = p+ans[p];
                    } else {    //相等时
                        if (ans[p] != 0) {  //相等元素的后边位置存在比它大的
                            ans[i] = ans[p] + p - i;
                        } else {    //相等元素在它之后没有小于它的值
                            ans[i] = 0;
                        }
                        break;
                    }
                }
            } else {    //这边也要判断
                if(ans[i+1] != 0) ans[i] = ans[i+1] + 1;
                else ans[i] = 0;
            }
        }
        return ans;
    }
}

代码2(Stack):

class Solution {
    public int[] dailyTemperatures(int[] T) {
        Stack<Integer> stk = new Stack<Integer>();
        int[] ans = new int[T.length];
        for (int i = 0; i < T.length; i++) {
            while (!stk.empty() && T[stk.peek()] < T[i]) {
                ans[stk.peek()] = i - stk.peek();
                stk.pop();
            }
            stk.push(i);    //因为无论与栈顶元素的比较结果如何,当前i位置的元素肯定要进栈。
        }
        return ans;

    }
}

代码2虽然短,但耗时较长。
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值