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

题目来源:https://leetcode-cn.com/problems/binary-subarrays-with-sum/

大致题意:
给定一个由0和1构成的数组和一个goal,判断数组中有多少连续的子序列和为goal。

思路

若子序列和为goal,且子序列起始索引为 i 和 j。那么从索引 0 至 i-1 的所有元素和为sub, 从索引 0 至 j 的所有数为sum。可知它们的关系为:

  • sub + goal = sum

于是使用一个值sum保存从第一个元素至今所有数组元素和,然后使用sum减去goal得到差值sub,然后使用哈希表获取sub的次数(即当前位置之前,数组有多个前缀序列的和为sub),那么sub的次数也就代表了goal的次数(也就是从数组首位至今,有多少个尾部为当前元素的子序列的序列和为goal)。然后将本轮次数加到结果中。

需要注意,如果子序列起始位为0,那么sub值也就为0,然后在map中查找0可能无法找到,导致本次结果无法统计。所以需要在遍历前在map中加入一个(0,1)对。

这样的时空复杂度都为O(n)。

升级思路

对于序列和为goal的子序列,它的开头和尾部可能是由0组成,每个0都会让统计次数加1,所以如果能直接统计序列开头和尾部的0的个数,然后滑动窗口,则可以在时间复杂度不变的情况下,达到O(1)的空间复杂度。

但是我自己写了一遍好多bug,看题解用了两个滑动窗口,窗口一存的是以当前遍历元素为结尾,元素和为goal的最长的子序列,起始位为left1;窗口二存的是以当前遍历元素为结尾的元素和刚好小于goal(也就是goal-1)的子序列,起始位为left2。然后left2-left1就是本轮统计结果。其实这个值刚好等于 从left1(left1 < 当前位置)位置开始连续的0元素个数 + 1。

代码:

public int numSubarrayWithSum(int[] nums, int goal) {
        int num = 0;
        // 方法一
        // Map<Integer, Integer> sumMap = new HashMap<Integer, Integer>();
        // int tempSum = 0;
        // sumMap.put(0, 1); // 初始化,这样保证第一个子序列和为goal的序列也可以被统计上
        // for(int i = 0; i < nums.length; i++) {
        //     tempSum += nums[i]; // 当前总的序列和
        //     int subNum = tempSum - goal; // 减去goal得差值
        //     int count = sumMap.getOrDefault(subNum, 0); // 判断在此之前有多少个差值,若无则为0
        //     num += count; // 统计
        //     sumMap.put(tempSum, sumMap.getOrDefault(tempSum, 0) + 1); // 将当前序列和放入map
        // }
        
        // 方法二
        int left1, left2;
        int right; // 当前位置
        int sum1, sum2;
        left1 = left2 = right = sum1 = sum2 = 0;
        
        while (right < nums.length) {
            sum1 += nums[right];
            while (left1 <= right && sum1 > goal) { // 窗口一
                sum1 -= nums[left1++];
            }
            sum2 += nums[right];
            while (left2 <= right && sum2 >= goal) { // 窗口二
                sum2 -= nums[left2++];
            }
            num += left2 - left1;
            right++;
        } 
        return num;
    }
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

三更鬼

谢谢老板!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值