单调栈解题模板

单调栈简介

单调栈是在栈（先进后出）的基础上，进一步地对入栈的数据进行控制，使得栈中的元素以递减，或者递增地顺序排列，单调栈适合解决，寻找下一个更大（更小）的元素这种问题，力扣上这种题型基本都是使用单调栈求解的。一般情况下，我们应该将值的索引入栈，而不是将值入栈，因为将索引入栈，既可以得到索引信息，也可以通过索引得到值的信息。

代码模板

我们既可以逆序遍历，也可以顺序遍历，labuladong前辈的代码是后序遍历模板，这里我的模板参考了力扣官方题解，使用正序遍历
以下模板是使用了循环数组的模板，如果不需要使用循环数组进行循环搜索，则i<n即可。

    for i:=0; i<2*n-1; i++ {
        for len(stack) != 0 && nums[stack[len(stack)-1]] < nums[i%n] {
            res[stack[len(stack)-1]] = nums[i%n]
            stack = stack[:len(stack)-1]
        }
        stack = append(stack, i%n)
    }

题目

下一个更大元素II

输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数；
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

func nextGreaterElements(nums []int) []int {
    n := len(nums)
    stack, res := make([]int, 0), make([]int, n)
    for i:=0; i<n; i++ {
        res[i] = -1
    }

    for i:=0; i<2*n-1; i++ {
        for len(stack) != 0 && nums[stack[len(stack)-1]] < nums[i%n] {
            res[stack[len(stack)-1]] = nums[i%n]
            stack = stack[:len(stack)-1]
        }
        stack = append(stack, i%n)
    }

    return res
}

下一个更大元素I
https://leetcode-cn.com/problems/next-greater-element-i/
这道题需要使用哈希表，key为nums[2]中的某个数，value为key对应的值的下一个更大的元素
此题的解法也是套模板，并参考了官方题解，先把结果都存储在map里面，然后返回需要的res

func nextGreaterElement(nums1 []int, nums2 []int) []int {
    m := make(map[int]int)

    res, stack := make([]int, len(nums1)), make([]int, 0)

    for i:=0; i<len(nums2); i++ {
        for len(stack) != 0 && nums2[stack[len(stack)-1]] < nums2[i] {
            k, v := nums2[stack[len(stack)-1]], nums2[i]
            m[k] = v
            stack = stack[:len(stack)-1]
        }
        stack = append(stack, i)
    }

    for j:=0; j<len(nums1); j++ {
        if _, ok := m[nums1[j]]; !ok {
            res[j] = -1
        }else {
            res[j] = m[nums1[j]]
        }
    }

    return res
}

leetcode739:每日温度

func dailyTemperatures(temperatures []int) []int {
    stack := make([]int, 0)
    res := make([]int, len(temperatures))

    for i:=0; i<len(temperatures); i++ {
        for len(stack)!=0 && temperatures[stack[len(stack)-1]] < temperatures[i] {
            res[stack[len(stack)-1]] = i - stack[len(stack)-1]
            stack = stack[:len(stack)-1]
        } 
        stack = append(stack, i)
    }

    return res
}