算法：单调栈的奥秘

最新推荐文章于 2024-10-01 23:05:47 发布

master_chenchengg

最新推荐文章于 2024-10-01 23:05:47 发布

阅读量300

点赞数 3

分类专栏：算法提升文章标签：算法 C++ 思维提升链表

本文链接：https://blog.csdn.net/master_chenchen/article/details/140190475

版权

算法提升专栏收录该内容

97 篇文章 0 订阅

订阅专栏

算法：单调栈的奥秘

引言

在算法的广阔天地里，C++如同一把锋利的工具，为开发者们开辟了一条通向高效数据处理的道路。而在众多数据结构中，单调栈犹如一颗璀璨的宝石，以其独特的魅力吸引着无数算法爱好者。本文旨在通过生动有趣的叙述，带你探索单调栈的神秘世界，了解其核心原理与实战应用，让你在算法的征途中增添一份利器。

技术概述

单调栈，顾名思义，是一种特殊的栈数据结构，其中元素保持单调递增或单调递减的顺序。这种数据结构在处理与局部最大值或最小值相关的问题时，展现出了非凡的能力。其核心特性包括：

单调性：栈中元素遵循严格递增或递减的顺序。
高效查询：能够快速找到栈中最大或最小元素。
灵活性：支持快速插入和删除操作，同时保持单调性不变。

代码示例

下面是一个简单的C++代码示例，展示了如何实现一个单调递增栈，并用于找出数组中的下一个更大元素。

#include <vector>
#include <stack>

std::vector<int> nextGreaterElement(std::vector<int>& nums) {
    std::vector<int> result(nums.size(), -1);
    std::stack<int> monoStack;
    
    for (int i = 0; i < nums.size(); ++i) {
        while (!monoStack.empty() && nums[monoStack.top()] < nums[i]) {
            result[monoStack.top()] = nums[i];
            monoStack.pop();
        }
        monoStack.push(i);
    }
    
    return result;
}

技术细节

单调栈的实现原理在于，通过维护一个单调的栈，能够在遍历数据的过程中，快速定位到局部的最大值或最小值。当新元素加入时，栈顶元素若不满足单调性，则持续弹出栈顶元素，直至满足条件或栈为空，再将新元素压入栈中。这样的操作保证了栈内元素的单调性，同时也为后续的查询提供了便利。

实战应用

单调栈在多种场景下都有着广泛的应用，尤其在股票交易、图像处理、信号分析等领域。例如，通过单调栈可以高效地计算股票价格的波动趋势，找出连续上涨或下跌的区间，为投资者提供决策依据。

代码示例（股票价格波动趋势）

#include <vector>
#include <stack>

std::vector<int> stockSpan(std::vector<int>& prices) {
    std::vector<int> spans(prices.size());
    std::stack<int> monoStack;
    
    for (int i = 0; i < prices.size(); ++i) {
        while (!monoStack.empty() && prices[monoStack.top()] <= prices[i]) {
            monoStack.pop();
        }
        spans[i] = monoStack.empty() ? i + 1 : i - monoStack.top();
        monoStack.push(i);
    }
    
    return spans;
}

优化与改进

尽管单调栈在处理局部最大值或最小值问题时表现优秀，但在某些极端情况下，如数据流中频繁出现与栈顶元素相同值的情况，可能会导致栈的大小异常增长，影响性能。为了解决这一问题，可以引入额外的数据结构，如哈希表，辅助记录元素的频次，从而在保证单调性的同时，避免栈的过度膨胀。

代码示例（优化后的单调栈）

#include <vector>
#include <stack>
#include <unordered_map>

std::vector<int> optimizedNextGreaterElement(std::vector<int>& nums) {
    std::vector<int> result(nums.size(), -1);
    std::stack<int> monoStack;
    std::unordered_map<int, int> freqMap;
    
    for (int i = 0; i < nums.size(); ++i) {
        while (!monoStack.empty() && nums[monoStack.top()] < nums[i]) {
            result[monoStack.top()] = nums[i];
            monoStack.pop();
        }
        freqMap[nums[i]]++;
        monoStack.push(i);
    }
    
    return result;
}