leetcode 930. 和相同的二元子数组

萌の鱼2025-01-28 19:27

题目如下

数据范围

示例

复制代码

在这道题之前我们曾经写过 1358. 包含所有三种字符的子字符串数目(以下称1358)。1358实际上就是让我们找到最小符合题意的子串,那么跟这道题有什么关系呢？
如果我把求和相同的数目改为求大于等于的数目呢？
显然直接求等于的数量不好求那么我们可以用大于等于的数量减去大于的数量那不就是等于的数量么。
所以我们的方法还是一样：不定长的滑动窗口固定右端点移动左端点来寻找最小符合题意的子数组。

通过代码
1358 寻找最小符合题意的子串可以看看这个

cpp 复制代码

class Solution {
public:
    int numSubarraysWithSum(vector<int>& nums, int goal) {
        int n = nums.size();
        int sum = 0;
        int count = 0;
        for(int i = 0,j = 0;i < n;i++){
            sum += nums[i];
            while(j <= i && sum > goal)sum -= nums[j++];
            count -= j;
        }sum = 0;
           for(int i = 0,j = 0;i < n;i++){
            sum += nums[i];
            while(j <= i && sum >= goal)sum -= nums[j++];
            count += j;
        }
        return count;
    }
};