单调栈的运用(包含0值哨兵+前缀和)

LeetCode1856子数组最小乘积的最大值

首先什么是单调栈

单调栈的存储的元素是按升序或逆序存储,所以叫做单调栈。
根据题意我们要求出对于已知数组中该元素左边第一个小于它的元素和右边第一个小于它的元素。
这就需要运用到单调栈来实现。

0值哨兵

第0个元素左边是没有数值的,为了处理边界,我们要在已知数组左边和右边加上一个0值哨兵。
class Solution {
public:
    int maxSumMinProduct(vector<int>& nums) {
        // 左右添加两个哨兵,方便单调栈内的判断
        nums.insert(nums.begin(), 0);
        nums.push_back(0);
        // 前缀和
        vector<long long> presum = {0};
        for(auto& n: nums)
            presum.push_back(presum.back() + n);
        
        // 右边第一个比它小的元素下标
        stack<int> s; //栈中存储的为下标。
        vector<int> rightFirstSmaller(nums.size(), 0);  //存储的也为下标。
        for(int i = 0;i < nums.size();i++){
            // 如果当前元素比栈顶元素小,弹栈
            while(!s.empty() && nums[i] < nums[s.top()]){
                int index = s.top();
                s.pop();
                rightFirstSmaller[index] = i;
            }
            s.push(i);
        }
        
        // 左边第一个比它小的元素下标
        s = stack<int>();
        vector<int> leftFirstSmaller(nums.size(), 0);
        for(int i = nums.size()-1;i >= 0;i--){
            // 如果当前元素比栈顶元素小,弹栈
            while(!s.empty() && nums[i] < nums[s.top()]){
                int index = s.top();
                s.pop();
                leftFirstSmaller[index] = i;
            }
            s.push(i);
        }
        
        // 比较每个元素的最大结果,打擂台得到答案
        long long res = 0;
        for(int i = 1;i < nums.size()-1;i++){
            int l = leftFirstSmaller[i];
            int r = rightFirstSmaller[i];
            res = max(res,nums[i] * (presum[r] - presum[l+1]));
        }
        return res % 1000000007;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值