day50| ● 739. 每日温度 ● 496.下一个更大元素 I

最新推荐文章于 2025-12-21 15:46:06 发布

原创最新推荐文章于 2025-12-21 15:46:06 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#哈希算法 #算法

数据结构与算法专栏收录该内容

54 篇文章

订阅专栏

文章介绍了如何使用单调栈解决每日温度问题和下一个更大元素问题，通过单调栈找到数组中元素的左右边界，时间复杂度均为O(n)。涉及递增和递减栈的使用以及hash映射辅助处理两个数组的情况。

739. 每日温度

到了单调栈的环节，要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置，此时我们就要想到可以用单调栈了。时间复杂度为O(n)。
如果求一个元素右边第一个更大元素，单调栈就是递增的（从栈头到栈底），如果求一个元素右边第一个更小元素，单调栈就是递减的。
而且还不要忘记，当判断完当前的后，要把当前i也压到栈里。

func dailyTemperatures(temperatures []int) []int {
    res := make([]int, len(temperatures))
    stack := make([]int, 0)
    stack = append(stack, 0)
    for i := 1; i < len(temperatures); i++{
        if temperatures[i] > temperatures[stack[len(stack) - 1]]{
            for len(stack) != 0 && temperatures[i] > temperatures[stack[len(stack) - 1]]{
                res[stack[len(stack) - 1]] = i - stack[len(stack) - 1]
                stack = stack[:len(stack) - 1]
            }
            stack = append(stack, i)
        }else{
            stack = append(stack, i)
        } 
    }
    return res
}

496.下一个更大元素 I

首先还是单调栈的思想，但是由于是两个数组，所以需要用hash建立映射。

func nextGreaterElement(nums1 []int, nums2 []int) []int {
    hash := make(map[int]int)
    for i := 0; i < len(nums1); i++{
        hash[nums1[i]] = i
    }

    stack := make([]int, 0)
    stack = append(stack,0)
    tmp := make([]int, len(nums1))
    for i := 0; i < len(tmp); i++{
        tmp[i] = -1
    }
    for i := 1; i < len(nums2);i++{
        if nums2[i] > nums2[stack[len(stack) - 1]]{
            for len(stack) != 0 && nums2[i] > nums2[stack[len(stack) - 1]]{
                if v, ok := hash[nums2[stack[len(stack) - 1]]];ok{
                    tmp[v] = nums2[i]
                }
                stack = stack[:len(stack) - 1]
            }
            stack = append(stack, i)
        }else{
            stack = append(stack, i)
        }
    }
    return tmp
}