Leetcode-930. 和相同的二元子数组

Problem: 930. 和相同的二元子数组

思路

滑动窗口

解题过程

我们可以通过计算 和大于等于 goal 的子数组数目和大于等于 goal+1 的子数组数目差值,来得到 和恰好等于 goal 的子数组数目。

Code

c++

cpp 复制代码
class Solution {
public:
    int at_most(vector<int>& nums, int goal) {
        int ans = 0;
        int n = nums.size();
        int l = 0;
        int sum = 0;
        for(int r = 0; r < n; r ++){
            sum += nums[r];
            while(sum >= goal && l <=r){
                sum -= nums[l];
                l++;
            }
            ans += l;
        }
        return ans;
    }
    int numSubarraysWithSum(vector<int>& nums, int goal) {
        return at_most(nums, goal) - at_most(nums,goal+1);
    }
};

python

python 复制代码
class Solution:

    def numSubarraysWithSum(self, nums: List[int], goal: int) -> int:
        def at_most(goal):
            ans = 0
            n = len(nums)
            l = 0
            sum = 0
            for r, x in enumerate(nums):
                sum += x
                while sum >= goal and l <= r:
                    sum -= nums[l]
                    l += 1
                ans += l
            return ans

        return at_most(goal) - at_most(goal + 1)

复杂度

  • 时间复杂度: O(n)O(n) O(n)
  • 空间复杂度: O(1)O(1) O(1)
相关推荐
想唱rap3 分钟前
直接选择排序、堆排序、冒泡排序
c语言·数据结构·笔记·算法·新浪微博
老葱头蒸鸡1 小时前
(27)APS.NET Core8.0 堆栈原理通俗理解
算法
视睿1 小时前
【C++练习】06.输出100以内的所有素数
开发语言·c++·算法·机器人·无人机
柠檬07112 小时前
matlab cell 数据转换及记录
算法
YuTaoShao3 小时前
【LeetCode 每日一题】2221. 数组的三角和
数据结构·算法·leetcode
little~钰3 小时前
树上倍增和LCA算法---上
算法
力扣蓝精灵3 小时前
今日分享 整数二分
算法
mc23563 小时前
5分钟学会微算法——Brian Kernighan 算法
算法
Excuse_lighttime3 小时前
除自身以外数组的乘积
java·数据结构·算法·leetcode·eclipse·动态规划
万添裁3 小时前
归并排序的三重境界
数据结构·算法