九章算法 | 拼多多面试题:和相同的二元子数组

描述

在由若干 0 和 1 组成的数组 A 中,有多少个和为 S 的非空子数组。

  • A.length <= 30000
  • 0 <= S <= A.length
  • A[i] 为 0 或 1

在线评测地址

LintCode 领扣

样例1

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

样例2

Input: A = [0,0,0,0,0,0,1,0,0,0], S = 0 
Output: 27 
Explanation:  
And 27 subarrays for S. 

题解

前缀和:定义一个数组sumN+1,si表示数组A中前i个元素之和,然后遍历sum数组,计算si+S(含义:前i个元素之和是si,找和为S的子数组个数)。求si+S的个数

public class Solution { 
    /** 
     * @param A: an array 
     * @param S: the sum 
     * @return: the number of non-empty subarrays 
     */ 
    public int numSubarraysWithSum(int[] A, int S) { 
        // Write your code here. 
        if (A == null || A.length == 0) { 
            return 0; 
        } 
        int prefixSum = 0; 
        int res = 0; 
        // counts[i] means the number of prefixSum = i 
        int[] counts = new int[A.length + 1]; 
        counts[0] = 1; 
        for (int i = 0; i < A.length; i++) { 
            prefixSum += A[i]; 
            if (prefixSum >= S) { 
                res += counts[prefixSum - S]; 
            } 
            counts[prefixSum]++; 
        } 
        return res; 
    } 
} 
 

更多题解参考:九章算法

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值