• Leetcode 795. 区间子数组个数


    好久没更新啦~最近忙着课题呢 研究生生活感觉有点子没意思 多学习总结 热爱生活!!芜湖~!

    题目描述:

    给你一个整数数组 nums 和两个整数:left 及 right 。找出 nums 中连续、非空且其中最大元素在范围 [left, right] 内的子数组,并返回满足条件的子数组的个数。

    生成的测试用例保证结果符合 32-bit 整数范围。

    题目解析:

    一看到这道题我第一反应就是,单调栈,左数组记录距离自己左边最近的大于自己的数,右数组记录距离自己右边最近的大于等于自己的数,这里为什么左边是大于,右边是大于等于,构造一个半闭半开的区间来进行自动去重,然后计算每个数对于结果的贡献量,即(当前序号减去左数组对应序号)乘上(右数组对应序号减去当前序号),参考代码可能会好理解一点,代码如下:

    1. class Solution {
    2. public:
    3. int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
    4. vector<int> leftarr(nums.size(),-1);
    5. vector<int> rightarr(nums.size(),nums.size());
    6. stack<int> stkl;
    7. stack<int> stkr;
    8. for(int i = 0 ; isize() ; i++){
    9. if(stkl.empty()){
    10. stkl.push(i);
    11. continue;
    12. }
    13. while(!stkl.empty()&&nums[stkl.top()]<=nums[i]){
    14. stkl.pop();
    15. }
    16. if(!stkl.empty())leftarr[i] = stkl.top();
    17. stkl.push(i);
    18. }
    19. for(int i = rightarr.size()-1 ; i>=0 ; i--){
    20. if(stkr.empty()){
    21. stkr.push(i);
    22. continue;
    23. }
    24. while(!stkr.empty()&&nums[stkr.top()]
    25. stkr.pop();
    26. }
    27. if(!stkr.empty())rightarr[i] = stkr.top();
    28. stkr.push(i);
    29. }
    30. int res = 0;
    31. for(int i = 0 ; isize() ; i++){
    32. if(nums[i]<=right&&nums[i]>=left)res += (i-leftarr[i])*(rightarr[i]-i);
    33. }
    34. return res;
    35. }
    36. };

    但是时间运行结果不太理想,在看了答案过后突然发现,使用双指针可以解决这个问题。

    我把数据分为三类,第一类小于left,第二类在left和right的全闭区间内,第三类大于right。我遍历数组,以当前遍历点为子数组的右区间,记录上一次出现第二类点的位置和上一次出现第三类点的位置,那么子数组的左区间可以在(last3,last2],将last2-last3加在答案中即可,代码如下:

    1. class Solution {
    2. public:
    3. int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
    4. int last2 = -1 , last1 = -1;
    5. int res = 0;
    6. for(int i = 0 ; i < nums.size() ; i++){
    7. if(nums[i]<=right&&nums[i]>=left){
    8. last1 = i;
    9. }
    10. else if(nums[i]>right){
    11. last1 = -1;
    12. last2 = i;
    13. continue;
    14. }
    15. if(last1!=-1)res += last1-last2;
    16. }
    17. return res;
    18. }
    19. };

    执行用时:64 ms, 在所有 C++ 提交中击败了31.52%的用户

    内存消耗:51.1 MB, 在所有 C++ 提交中击败了30.37%的用户

    目标值为最大值在left和right之间的子数组数,可以转换为最大值小于等于right的子数组数减去最大值小于left的子数组数,一次遍历进行计算,代码如下:

    1. class Solution {
    2. public:
    3. int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
    4. return cal(nums,right)-cal(nums,left-1);
    5. }
    6. int cal(vector<int>& nums , int num){
    7. int res = 0 , cur = 0 ;
    8. for(int i = 0 ; isize() ; i++){
    9. cur = nums[i]<=num?cur+1:0;
    10. res += cur;
    11. }
    12. return res;
    13. }
    14. };

    执行用时:56 ms, 在所有 C++ 提交中击败了61.03%的用户

    内存消耗:51 MB, 在所有 C++ 提交中击败了48.42%的用户

    感觉自己coding有进步,自动去重单调栈一次就写出来了,多加油,科研去了!

  • 相关阅读:
    Xib添加scrollView再添加subView约束不报错的两种方式
    Linux-缓冲区(简单理解)
    在农业数据集上运行VINS-fusion
    Vue入门
    线性代数中涉及到的matlab命令-第一章:行列式
    2022年跨境电商卖家如何在Facebook上做广告【完整指南】
    Servlet规范之安全
    Python正则表达式
    Redis常见异常汇总
    在“百模大战”重生,搜索引擎又行了?
  • 原文地址:https://blog.csdn.net/rygy_/article/details/128014362