Day04 前缀和&差分 1109. 航班预订统计 、304. 二维区域和检索 - 矩阵不可变

发布于:2025-09-14 ⋅ 阅读:(21) ⋅ 点赞:(0)
一、304. 二维区域和检索 - 矩阵不可变
1、思路(二维前缀和)

使用二维前缀合
在这里插入图片描述

2、代码
class NumMatrix {
    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 = 1; i <= m; i++) {
            for (int j = 1; j <= n; j++) {
                // 注意索引是从1开始,当前值索引记得 -1
                sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + matrix[i - 1][j - 1];
            }
        }
    }

    public int sumRegion(int row1, int col1, int row2, int col2) {
        //注意题目索引从0开始,公式索引是从1开始
        row1++;
        col1++;
        row2++;
        col2++;
        return sum[row2][col2] - sum[row2][col1 - 1] - sum[row1 - 1][col2] + sum[row1 - 1][col1 - 1];
    }
}

/**
 * Your NumMatrix object will be instantiated and called as such:
 * NumMatrix obj = new NumMatrix(matrix);
 * int param_1 = obj.sumRegion(row1,col1,row2,col2);
 */
二、1109. 航班预订统计
1、思路(差分&前缀和)

在这里插入图片描述

1、差分数组B的前缀和数组就是原数组A

B1+B2+....+ Bn = A1+ A2 -A1 + A3-A2+A4-A3... +An - An-1 =  An


B = delta(A;
A = sum(B)
即:A =sum(delta(A);

2、差分+前缀和 善于处理问题,对A数组 某连续端统一加 一个数字的问题

这类问题可以先得出差分,对差分计算,再求前缀和即可以得到答案

图中例子1拆解:

原数组A为[0,1,-1,-1,0] 需要 对 2 到 4 个数+1 最终应该为:【0,2,0,0,0】

1> 先计算出原数组A差分数组B为:【0,1,-2,0,1】,再对差分B相同位置 +1 

B2 = A2 -A1 ; ------------>A1没变,A2本身+1了,即b2+1

B3 = A3-A2;------------> A2和A3都加1,抵消,不变

B4 = A4-A3;------------>A4和A3都加1,抵消,不变

B5 = A5-A4;------------>A5没变,A4本身+1了,即B5 -1

(L为2,r为4,1为d)得出公式:	**Bₗ** + d  ,B(r+1) - d 

2> 按上面得出公式对差分数组B操作得到:【0,2,-2,0,0】

3>最后对差分数组B求前缀和得到结果:【0,2,0,0,0】
2、代码
class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] delta = new int[n + 2];  // 差分要开0~n+1
        Arrays.fill(delta, 0);
        for (int[] booking : bookings) {
            int fir = booking[0];
            int last = booking[1];
            int seats = booking[2];
            // 差分模板
            delta[fir] += seats;
            delta[last + 1] -= seats;
        }

        int[] sum = new int[n + 1]; // 0~n
        for (int i = 1; i <= n; i++) {
            sum[i] = sum[i - 1] + delta[i];
        }
        // 往前移一位,前缀和求出来的是索引为1开始的
        int[] result = new int[n];
        for (int i = 1; i <= n; i++) {
            result[i - 1] = sum[i];
        }
        return result;
    }
}