930. Binary Subarrays With Sum

最新推荐文章于 2021-07-08 09:30:36 发布

Simple_ R

最新推荐文章于 2021-07-08 09:30:36 发布

阅读量171

点赞数

分类专栏： LeedCode365 文章标签： leetcode

本文链接：https://blog.csdn.net/M_sdn/article/details/83871149

版权

LeedCode365 专栏收录该内容

68 篇文章 0 订阅

订阅专栏

本文探讨了在一个由0和1组成的数组A中，寻找所有和为S的非空子数组的问题。通过一个示例说明，当输入数组为[1,0,1,0,1]且目标和为2时，存在4个符合条件的子数组。文章提供了详细的算法实现，展示了如何遍历数组并计算子数组和，最终返回满足条件的子数组数量。

摘要由CSDN通过智能技术生成

In an array A of 0s and 1s, how many non-empty subarrays have sum S?

Example 1:

Input: A = [1,0,1,0,1], S = 2
Output: 4
Explanation: 
The 4 subarrays are bolded below:
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]

Note:

A.length <= 30000
0 <= S <= A.length
A[i] is either 0 or 1.

public int numSubarraysWithSum(int[] A, int S) {
        int res = 0;
        for (int i = 0; i < A.length; i++) {
            int sum = 0, j = i;
            if (A[j] > S) continue;
            if (sum == S) j++;
            while (sum < S) {
                if (j < A.length) {
                    sum += A[j++];
                } else {
                    break;
                }
            }
            if (sum == S) res += 1;
            try{
                while (A[j] == 0) {
                    res += 1;
                    j++;
                }
            } catch (Exception e) {
                continue;
            }
        }
        return res;
    }