Monotonic stack

来自wrc's Wiki
跳到导航 跳到搜索

单调栈。用于处理 Next Greater Element 这类问题。

相关题目

LeetCode 496. Next Greater Element I

vector<int> nextGreaterElement(vector<int>& nums) {
    vector<int> res(nums.size()); // 存放答案的数组
    stack<int> s;
    // 倒着往栈里放
    for (int i = nums.size() - 1; i >= 0; i--) {
        // 判定个子高矮
        while (!s.empty() && s.top() <= nums[i]) {
            // 矮个起开,反正也被挡着了。。。
            s.pop();
        }
        // nums[i] 身后的 next great number
        res[i] = s.empty() ? -1 : s.top();
        // 
        s.push(nums[i]);
    }
    return res;
}

LeetCode 739. Daily Temperatures

变种,需要返回和 next greater element 之间的距离。

def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
    ret = [0 for _ in range(len(temperatures))]
    stack = []
    for i in reversed(range(len(temperatures))):
        while stack and temperatures[stack[-1]] <= temperatures[i]:
            stack.pop()
        ret[i] = 0 if not stack else stack[-1] - i
        stack.append(i)
    return ret

另见