前缀和总结

前缀和总结

• 通常是利用一个数组或集合记录原数组中前n项和S_n
• 通过S_n - S_n-i可以得到任意子区间的和
• 需要注意的是根据题意不同，这个“和”也有可能是次数相加得到的和，不一定就是数组值相加得到的和，需要灵活应对。

通常情况下，我们可以利用前缀和思想进行降维，将时间复杂度O（n²）降到O（n）

以力扣中题号560这道题为例
题目：
给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的连续子数组的个数 。

示例1：

输入：nums = [1,1,1], k = 2
输出：2

答案：

class Solution {
    public int subarraySum(int[] nums, int k) {
        //键：以i为右边界的区间和
        //值：出现的次数
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        hashMap.put(0,1);
        int cnt = 0;
        int sum = 0;
        for (int i=0;i<nums.length;i++){
            sum += nums[i];
            cnt += hashMap.getOrDefault(sum - k,0);
            if (hashMap.containsKey(sum))hashMap.put(sum, hashMap.get(sum) + 1);
            else hashMap.put(sum,1);
        }
        return cnt;
    }
}

• 这题里我们可以暴力法解决，即使用双重循环解决，但是内循环我们可以利用前缀和来优化
• 如果此时数组前n项和S_n=n，如果存在和为k的子区间，那么该子区间个数等于和为n-k的区间个数
• 为了能够在O（1）内得到和为n-k的区间个数，我们可以利用哈希表来记录，键是S_n，值则是个数
• 其中的hashMap.put(0,1) 很关键，如果S_n=k，那么n-k=0，也就是要把区间自身就等于k这个情况考虑进去
• 我们应该先统计个数，再将S_n记录进哈希表里，以{-1,1}，k=0这个情况为例，如果反过来的话那么得到的结果就是2了，因为我们已经提前将0-1记录进哈希表里了