前缀和算法

1.相关定义

        

对于一个给定的数列 𝐴 ，它的前缀和数列 𝑆 中 𝑆[𝑖] 表示从第 1 个元素到第 𝑖 个元素的总和。用公式表示为：

𝑆[𝑖]=∑𝑗=1𝑖𝐴[𝑗]

2.常见用法

       非常快的时间求出多个区间的和。每个查询的时间复杂度是 𝑂(1)。        

3.具体案例

        a.一维元素区间求和

给你一个数组 nums 。数组「动态和」的计算公式为：runningSum[i] = sum(nums[0]…nums[i]) 。

请返回 nums 的动态和。

class Solution {
    public int[] runningSum(int[] nums) {
        int n = nums.length;
        for (int i = 1; i < n; i++) {
            nums[i] += nums[i - 1];
        }
        return nums;
    }
}

        b.二维元素区间求和

                与一维前缀和的原理基本，只不过二维前缀和求的是一个矩阵中所有元素的和。

 例如上图：我们要求蓝色矩阵中所有元素的和。

class NumMatrix {
    private final int[][] sum;

    public NumMatrix(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        sum = new int[m + 1][n + 1];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                sum[i + 1][j + 1] = sum[i + 1][j] + sum[i][j + 1] - sum[i][j] + matrix[i][j];
            }
        }
    }

    // 返回左上角在 (r1,c1) 右下角在 (r2,c2) 的子矩阵元素和
    public int sumRegion(int r1, int c1, int r2, int c2) {
        return sum[r2 + 1][c2 + 1] - sum[r2 + 1][c1] - sum[r1][c2 + 1] + sum[r1][c1];
    }
}