C++前缀和算法全解

前缀和算法简介

前缀和算法（Prefix Sum）是一种用于快速计算数组区间和的预处理技术。通过预先计算并存储数组的前缀和，可以在常数时间内查询任意区间的和，适用于频繁查询的场景。

前缀和数组的构建

给定一个数组 nums，其前缀和数组 prefix 的定义为：

• prefix[0] = 0
• prefix[i] = nums[0] + nums[1] + ... + nums[i-1]（对于 i > 0）

构建前缀和数组的代码实现如下：

vector<int> buildPrefixSum(const vector<int>& nums) {
    int n = nums.size();
    vector<int> prefix(n + 1, 0);
    for (int i = 1; i <= n; ++i) {
        prefix[i] = prefix[i - 1] + nums[i - 1];
    }
    return prefix;
}

区间和查询

通过前缀和数组，可以快速计算原数组中任意区间 [l, r] 的和：

• 区间和 sum[l..r] = prefix[r + 1] - prefix[l]

查询示例代码：

int rangeSum(const vector<int>& prefix, int l, int r) {
    return prefix[r + 1] - prefix[l];
}

应用场景

1. 频繁区间和查询：如统计子数组的和。
2. 滑动窗口优化：结合哈希表解决子数组和为特定值的问题。
3. 动态规划预处理：用于优化状态转移中的区间计算。

示例问题

问题：给定数组 nums 和整数 k，统计和为 k 的子数组数量。

解法：利用前缀和与哈希表记录出现次数：

int subarraySum(vector<int>& nums, int k) {
    unordered_map<int, int> prefixCount;
    prefixCount[0] = 1;
    int sum = 0, result = 0;
    for (int num : nums) {
        sum += num;
        if (prefixCount.find(sum - k) != prefixCount.end()) {
            result += prefixCount[sum - k];
        }
        prefixCount[sum]++;
    }
    return result;
}

注意事项

• 数组越界：前缀和数组通常多分配一个元素（n + 1 大小）。
• 负数处理：若数组包含负数，前缀和可能非单调递增。
• 空间复杂度：前缀和算法需要额外的 O(n) 空间。