leetcode.795 区间子数组个数 - 单调栈 + 枚举元素计算贡献

795. 区间子数组个数

思路：

• 枚举当每个元素nums[i] 作为子数组的最大值时，统计以nums[i]为最大值的子数组的个数
• 以该元素为最大值的子数组的个数 = 
• l[i] 为 —— nums[i]左侧第一个大于等于该元素的下标
• r[i]为 —— nums[i]右侧第一个大于该元素的下标
• 这里要注意：会出现重复计数的情况
• 这里的 l[i] 和 r[i] 采用单调栈求出

class Solution {
    public int numSubarrayBoundedMax(int[] nums, int a, int b) 
    {
        int n=nums.length;
        int[] l=new int[n];
        int[] r=new int[n];
        Arrays.fill(l,-1);
        Arrays.fill(r,n);
        Deque<Integer> stk=new ArrayDeque<>();
        for(int i=0;i<n;i++) //求左端比x大的数的下标
        {
            int x=nums[i];
            while(!stk.isEmpty()&&nums[stk.peek()]<=x) stk.pop();
            if(!stk.isEmpty()) l[i]=stk.peek();
            stk.push(i);
        }
        stk.clear();
        for(int i=n-1;i>=0;i--) //求右端大于等于x的数的下标
        {
            int x=nums[i];
            while(!stk.isEmpty()&&nums[stk.peek()]<x) stk.pop();
            if(!stk.isEmpty()) r[i]=stk.peek();
            stk.push(i);
        }
        int res=0;
        for(int i=0;i<n;i++) 
            if(a<=nums[i]&&nums[i]<=b) res+=(i-l[i])*(r[i]-i);
        return res;
        
    }
}