- 区间子数组个数
给定一个元素都是正整数的数组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;
}
};