LeetCode 795. 区间子数组个数--单调栈

795. 区间子数组个数

给定一个元素都是正整数的数组A ，正整数 L 以及 R (L <= R)。

求连续、非空且其中最大元素满足大于等于L 小于等于R的子数组个数。

例如 :
输入:
A = [2, 1, 4, 3]
L = 2
R = 3
输出: 3
解释: 满足条件的子数组: [2], [2, 1], [3].

注意:

L, R  和 A[i] 都是整数，范围在 [0, 10^9]。
数组 A 的长度范围在[1, 50000]。

题解

利用单调栈找出每个元素在区间[L,R]内最大，于是在这区间取值便可以得到题目要求的答案，但是需要注意数组同时出现两个相同数字的情况，比如[2,1,2,5,3]，如果用单调栈直接处理，第一个2和第3个2的区间[L,R]是一样的，但是我们计算的时候就会把一些重复的算进去，所以我们在单调栈里面做一些规避，让第一个2的区间为[0,2]，第二个2的区间为[1,2]，也就是说，针对每个数字nums[x]，左边的区间[L,x]内该数字肯定最大，右边的区间[x,R]可以允许出现和该数字一样大的数字，这样后面做运算的话就可以有效避免重复的情况。

AC代码

class Solution {
public:
typedef long long ll;
    int L[100010],R[100010];
    stack<int>q;
    void Stack_clear()//清空栈函数 
    {
        while(!q.empty())
        q.pop();
    }
    //这里数组以下标1开始
    void get_L(vector<int>arr)//得到左区间 
    {
        for(int i=0;i<arr.size();i++)
        L[i]=i;//初始化
        Stack_clear();
        for(int i=0;i<arr.size();i++)
        {
            if(q.empty()||arr[q.top()]>=arr[i])
            q.push(i);
            else
            {
                while(q.empty()==false&&arr[q.top()]<arr[i])
                {
                    L[i]=L[q.top()];
                    q.pop();
                }
                q.push(i);
            }
        }
    }
    void get_R(vector<int>arr)//得到右区间，其实就是把数组反向遍历的左区间求解 
    {
        for(int i=0;i<arr.size();i++)
        R[i]=i;//初始化
        Stack_clear();
        for(int i=arr.size()-1;i>=0;i--)
        {
            if(q.empty()||arr[q.top()]>arr[i])
            q.push(i);
            else
            {
                while(q.empty()==false&&arr[q.top()]<=arr[i])
                {
                    R[i]=R[q.top()];
                    q.pop();
                }
                q.push(i);
            }
        }
    }
    int numSubarrayBoundedMax(vector<int>& nums, int left, int right) 
    {
        get_L(nums);
        get_R(nums);
        int res=0;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]>=left&&nums[i]<=right)
            {
                res+=((i-L[i]) + (R[i]-i) + (i-L[i])*(R[i]-i));
                res++;//自己本身单独作为一个
            }
        }
        return res;
    }
};