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

最新推荐文章于 2024-10-01 05:04:32 发布

爱玩代码的勒勒

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量90

点赞数

分类专栏：刷题日记文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/tqtaylor/article/details/120968485

版权

刷题日记专栏收录该内容

16 篇文章 0 订阅

订阅专栏

这篇博客介绍了两种方法解决寻找整数数组中和为k的连续子数组的个数问题。方法一是通过预处理前缀和，但不适用于包含负数的情况；方法二是利用HashMap存储前缀和及其出现的次数，有效地处理了负数情况并提高了效率。

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

给定一个整数数组和一个整数 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;
    }
}