前缀和数组和差分数组

1. 前缀和概念

一维区间和：

前缀和可以简单理解为「数列的前 项的和」，是一种重要的预处理方式，能大大降低查询的时间复杂度

代码实现：

public static void main(String[] args) {
        int[] nums = { 1, 2, 3, 4, 5 };
        int[] pre = new int[nums.length];
        pre[0] = nums[0];
        for (int i = 1; i < nums.length; i++) {
            pre[i] = pre[i - 1] + nums[i];// //pre[i]表示num[0]-nums[i]的元素累加和
        }
        System.out.println(Arrays.toString(pre));
    }

给定区间[i,j], 求出数组nums[i]+....+nums[j]的和 结果为: pre[j]-pre[i]+nums[i]

二维区间和：

class NumMatrix {
    int [][]preSum;
    public NumMatrix(int[][] matrix) {
        int m=matrix.length;
        int n=matrix[0].length;
        preSum=new int[m+1][n+1];//多一行一列便于处理
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=n;j++)
            {
                //二维前缀和的计算方法
                //preSum[x][y]: 表示（0，0）->(x-1,y-1)区域内的元素之和
                preSum[i][j]=preSum[i-1][j]+preSum[i][j-1]-preSum[i-1][j-1]+matrix[i-1][j-1];
            }
        }

    }
    
    //获取区间(row1,col1)->(row2,col2)区间内的和
    //**红色区域元素之和=绿色区域元素之和-蓝色区域之和-黄色区域之和+粉色区域元素之和**
    public int sumRegion(int row1, int col1, int row2, int col2) {
        return preSum[row2+1][col2+1]-preSum[row1][col2+1]-preSum[row2+1][col1]+preSum[row1][col1];

    }
}

---

2. 差分数组

前缀和数组使用与数组中的元素不改变的情况，而差分数字与之相反，适用于数组元素经常改变的情况。 比如对原始数组nums区间[i,j]上的元素都进行递增或者递减操作，需要执行的次数是j-i+1次，如果修改次数比较多，时间复杂度就比较高了，使用差分数组可以使得时间复杂度降为O(1)

原始数组nums, 差分数组定义如下：

• diff[0]=nums[0]
• diff[i]=nums[i]-nums[i-1] i>=1

创建差分数组

public void createDiff(int []nums)
    {
       diff[0]=nums[0];//设置diff[0]=nums[0]，是因为后面我们需要通过差分数组反推出数组
       for(int i=1;i<nums.length;i++)
        diff[i]=nums[i]-nums[i-1];

    }

如果对区间[i,j]上的元素都加上val(val可以为正也可以为负), 需要执行的操作是diff[i]+=val; diff[j+1]-=val

diff[i]=nums[i]-nums[i-1] diff[i]+=val, 则说明nums[i]比nums[i-1]增加了val,
diff[j+1]=nums[j+1]-nums[j] nums[j]增加了val 为了不影响nums[j+1] 因此将nums[j+1]与nums[j]的差缩小val

区间增加

 public void increase(int i,int j,int val)
    {
        diff[i]+=val;
        if(j+1<nums.length)
            diff[j+1]-=val;
    }

还原原始数组：

 public void getResult()
    {
        nums[0]=diff[0];
        for(int i=1;i<nums.length;i++)
        {
            nums[i]=nums[i-1]+diff[i];
        }
    }