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

一、304. 二维区域和检索 - 矩阵不可变
1、思路(二维前缀和)

使用二维前缀合

2、代码
JAVA 复制代码
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

java 复制代码
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、代码
JAVA 复制代码
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;
    }
}
相关推荐
高晶4 小时前
无乘法器实现LC二阶滤波器:基于移位运算的快速算法与四种精度推导
算法·设计模式·程序员
ysbfsy5 小时前
大数计算思路
算法
const_qiu5 小时前
第四章 00:开篇 —— 从「会聊天」到「记得住、会干活、能协作」
java·开发语言·人工智能·python
Wang's Blog5 小时前
Java 项目实战: 交友项目 - 个人资料管理的更新接口实现与坑
java·开发语言·交友
高晶5 小时前
主从自动分配算法:平均值乘以设备数一半的选举基准值解析
算法·设计模式·程序员
Promising_GEO5 小时前
GEE实战02期:从卫星波段开始:利用 Sentinel-2 自己计算 NDVI
java·sentinel·遥感·空间分析
大文说跨境5 小时前
亚马逊多账号管理浏览器怎么选?云登/紫鸟/AdsPower 对比
java·前端·自动化
CoderYanger5 小时前
Java EE 进阶:4.Spring Web MVC 入门(第一弹)
java·开发语言·前端·spring·面试·java-ee·mvc
学电子她就能回来吗5 小时前
LeetCode Hot 100|普通数组专题学习笔记(零基础版)
笔记·算法·leetcode
用户老李5 小时前
反向传播计算梯度举例
算法