当前位置：首页 > news >正文

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

news 2025/9/14 8:09:00

一、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开始，当前值索引记得 -1sum[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 =  AnB = 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+1B3 = 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+1Arrays.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~nfor (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;}
}