前缀和

前缀和

前缀和是一个重要的预处理，能大大降低查询的时间复杂度。
前缀和多用于数组的处理，是一个数组某个索引之前（包括当前索引）的所有数组元素和。本文前缀和使用preSum[i]表示

处理一维数组

使用前缀和处理一维数组时，可以从前到后遍历数组，perSum[i]表示索引i前面的所有数组元素和，

1. 设置perSum数组长度为len+1，是使得preSum[i]表示的是i之前的数组元素和不包含i。也为了只求0是可以使用该方法，不必再进行特殊的处理。所有在使用sumRange方法时数组索引使用j+1的原因

代码如下：

class PrefixSum{
    int[] preSum;
    public void NumArray(int[] arr){
        int len = arr.length;
        preSum = new int[len+1];
        for (int i = 0; i < len; i++) {
            preSum[i+1] = preSum[i] + arr[i];
        }
    }
    public int sumRange(int i,int j){
        return preSum[j+1] - preSum[i];
    }
}

处理二维数组

处理二维数组是有两种方法，
方法一：
在预处理时，使用上面的方法求得每行数组的前缀和，之后在通过数值求得特定子数组的和。

class PrefixSum {
    int[][] perSum;

    public PrefixSum(int[][] arr) {
        int len = arr.length;
        if (len > 0) {
            int n = arr[0].length;
            perSum = new int[len][n + 1];
            for (int i = 0; i < len; i++) {
                for (int j = 0; j < n; j++) {
                    perSum[i][j + 1] = perSum[i][j] + arr[i][j];
                }
            }
        }
    }
    
    public int getSum(int row1, int col1, int row2, int col2) {
        int sum = 0;
        for (int i = row1; i <= row2; i++) {
            sum += perSum[i][col2 + 1] - perSum[i][col1];
        }
        return sum;
    }
}

方法二:通过自己画矩阵可以通过递推找到规律，求某个子矩阵的和相当于公式

perSum[i + 1][j + 1] = perSum[i][j + 1] + perSum[i + 1][j] - perSum[i][j] + arr[i][j];

class NumMatrix {
    int[][] perSum;

    public NumMatrix(int[][] arr) {
        int len = arr.length;
        if (len > 0) {
            int n = arr[0].length;
            perSum = new int[len + 1][n + 1];
            for (int i = 0; i < len; i++) {
                for (int j = 0; j < n; j++) {
                    perSum[i + 1][j + 1] = perSum[i][j + 1] + perSum[i + 1][j] - perSum[i][j] + arr[i][j];
                }
            }
        }
    }
    
    public int getSum(int row1, int col1, int row2, int col2) {
        return perSum[row2 + 1][col2 + 1] - perSum[row1][col2 + 1] - perSum[row2 + 1][col1] + arr[row1][col1];
    }
}

参考资料：leetcode-cn.com