leetcode 探索队列与栈 每日温度

题目

请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

分析

这个题第一个想法是双指针,即从左侧向右扫描,left指针和right指针,遇到right指向的元素比left大的,就计算隔的天数,然后left前进一步,right回到left+1处,判断并继续向前扫描。不过提交时候超时了。。。这其实相当于双层循环,暴力解法。

题解里,更为巧妙的解法是,单调栈。这是一种专门针对前后左右的,比如接雨水的问题,也是这种解法。
在栈里存储的是索引,然后遍历整个温度,遇到温度比栈里的大的,就计算两个索引之间的距离,并对结果集赋值,也就是栈里这个索引所在的结果,然后将其弹出。继续拿栈顶与当前温度比较。

解法

func dailyTemperatures(T []int) []int {

    // 当前栈顶的是73,压入一个74,1,
    // 当前栈顶的是74,压入一个75,1,
    // 当前栈顶是75,压入一个71,+1,当前计算的是75。
    // 当前栈顶的是71,压入一个69,+1
    // 当前栈顶的是69,压入一个72,+1,
    // 当前栈顶的是72,压入一个76,+1,结束,day=4,
    // 栈弹出4-1个,弹3个出来,76,72,69,栈顶是71,
    // 当前栈顶是71,压入一个69,+1,
    // 当前栈顶是69,压入一个72,+1,结束,day=2
    // 栈弹出2-1,弹一个出来72,栈顶是69
    // 当前栈顶是69,压入一个,
    // 遇到结尾,没有找到大的,day=0,弹出n-1个来。
    
    // 双指针超时
    // var result []int
    // for i, j:=0, 1; j < len(T);  {
    //     if T[i] < T[j] {
    //         result = append(result, j-i)
    //         i++
    //         j=i+1
    //     } else {
    //         if j == len(T) - 1 {
    //             result = append(result, 0)
    //             i++
    //             j=i+1
    //         } else {
    //             j++
    //         }
    //     }
    // }
    // // 最后还要补上一个最后一位没被比较的0
    // result = append(result, 0)
    
    var stack []int
    var result = make([]int, len(T))
    
    for i := 0; i < len(T); {
        // 当前温度大于栈顶索引对应的问题,则对应的栈顶索引出等待天数为i- stack.peek()
        if len(stack) > 0 && T[i] > T[stack[len(stack)-1]] {
            result[stack[len(stack)-1]] = i - stack[len(stack)-1]
            stack = stack[:len(stack)-1]
        } else {
            stack = append(stack, i)
            i++
        }
        // fmt.Printf("stack: %v, t: %d, index : %d, \n", stack, T[i], i)
    }
    
    return result
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值