给你一个整数数组 nums 和两个整数:left 及 right 。找出 nums 中连续、非空且其中最大元素在范围 [left, right] 内的子数组,并返回满足条件的子数组的个数。
生成的测试用例保证结果符合 32-bit 整数范围。
一看到这道题我第一反应就是,单调栈,左数组记录距离自己左边最近的大于自己的数,右数组记录距离自己右边最近的大于等于自己的数,这里为什么左边是大于,右边是大于等于,构造一个半闭半开的区间来进行自动去重,然后计算每个数对于结果的贡献量,即(当前序号减去左数组对应序号)乘上(右数组对应序号减去当前序号),参考代码可能会好理解一点,代码如下:
- class Solution {
- public:
- int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
- vector<int> leftarr(nums.size(),-1);
- vector<int> rightarr(nums.size(),nums.size());
- stack<int> stkl;
- stack<int> stkr;
- for(int i = 0 ; i
size() ; i++){ - if(stkl.empty()){
- stkl.push(i);
- continue;
- }
- while(!stkl.empty()&&nums[stkl.top()]<=nums[i]){
- stkl.pop();
- }
- if(!stkl.empty())leftarr[i] = stkl.top();
- stkl.push(i);
- }
- for(int i = rightarr.size()-1 ; i>=0 ; i--){
- if(stkr.empty()){
- stkr.push(i);
- continue;
- }
- while(!stkr.empty()&&nums[stkr.top()]
- stkr.pop();
- }
- if(!stkr.empty())rightarr[i] = stkr.top();
- stkr.push(i);
- }
- int res = 0;
- for(int i = 0 ; i
size() ; i++){ - if(nums[i]<=right&&nums[i]>=left)res += (i-leftarr[i])*(rightarr[i]-i);
- }
- return res;
- }
- };
但是时间运行结果不太理想,在看了答案过后突然发现,使用双指针可以解决这个问题。
我把数据分为三类,第一类小于left,第二类在left和right的全闭区间内,第三类大于right。我遍历数组,以当前遍历点为子数组的右区间,记录上一次出现第二类点的位置和上一次出现第三类点的位置,那么子数组的左区间可以在(last3,last2],将last2-last3加在答案中即可,代码如下:
- class Solution {
- public:
- int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
- int last2 = -1 , last1 = -1;
- int res = 0;
- for(int i = 0 ; i < nums.size() ; i++){
- if(nums[i]<=right&&nums[i]>=left){
- last1 = i;
- }
- else if(nums[i]>right){
- last1 = -1;
- last2 = i;
- continue;
- }
- if(last1!=-1)res += last1-last2;
- }
- return res;
- }
- };
执行用时:64 ms, 在所有 C++ 提交中击败了31.52%的用户
内存消耗:51.1 MB, 在所有 C++ 提交中击败了30.37%的用户
目标值为最大值在left和right之间的子数组数,可以转换为最大值小于等于right的子数组数减去最大值小于left的子数组数,一次遍历进行计算,代码如下:
- class Solution {
- public:
- int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
- return cal(nums,right)-cal(nums,left-1);
- }
-
- int cal(vector<int>& nums , int num){
- int res = 0 , cur = 0 ;
- for(int i = 0 ; i
size() ; i++){ - cur = nums[i]<=num?cur+1:0;
- res += cur;
- }
- return res;
- }
- };
执行用时:56 ms, 在所有 C++ 提交中击败了61.03%的用户
内存消耗:51 MB, 在所有 C++ 提交中击败了48.42%的用户
感觉自己coding有进步,自动去重单调栈一次就写出来了,多加油,科研去了!