一维前缀和 连续数组和为k

给定一个整数数组和一个整数 k ,请找到该数组中和为 k 的连续子数组的个数。

滑动窗口没办法解决有负数的情况

方法一:

预处理 前缀和 sum_ij = preSum[j] - preSum[i-1];

class Solution {
    public int subarraySum(int[] nums, int k) {
        //滑动窗口没办法解决有负数的情况
        //预处理 前缀和 sum_ij = preSum[j] - preSum[i-1];\
         int n = nums.length;
         int[] preSums = new int[n+1];
         for(int i=0;i<n;++i)
         {
             preSums[i+1] = preSums[i] + nums[i];
         }

         int ans = 0;

         for(int i=1;i<=n;++i)
         {
             for(int j=0;j<i;++j)
             {
                 if(preSums[i]-preSums[j]==k) ++ans;
             }

         }
         return ans;
    }

方法二

使用HashMap,sumToCount的哈希表,其中key值是sum,value是count,sum-k如果有则加上该count

class Solution {
    public int subarraySum(int[] nums, int k) {
        //滑动窗口没办法解决有负数的情况
        //预处理 前缀和 sum_ij = preSum[j] - preSum[i-1];\
        // int n = nums.length;
        // int[] preSums = new int[n+1];
        // for(int i=0;i<n;++i)
        // {
        //     preSums[i+1] = preSums[i] + nums[i];
        // }

        // int ans = 0;

        // for(int i=1;i<=n;++i)
        // {
        //     for(int j=0;j<i;++j)
        //     {
        //         if(preSums[i]-preSums[j]==k) ++ans;
        //     }

        // }
        // return ans;

        //使用map<sum,count>
        Map<Integer,Integer> sumCount = new HashMap<>();
        sumCount.put(0,1);
        int sum = 0;
        int count = 0;
        for(int num : nums)
        {
            sum += num;
            count += sumCount.getOrDefault(sum-k,0);//取不到就是0
            sumCount.put(sum, sumCount.getOrDefault(sum,0)+1);//sum的 次数加1
        }
        return count;
    }
}
posted @ 2021-10-26 11:18  勒勒乐了  阅读(61)  评论(0)    收藏  举报