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;
    }
}
相关推荐
卷福同学5 小时前
不用服务器,不用配环境,我10分钟上线了一个AI Agent
人工智能·后端·算法
码兄科技7 小时前
Java AI智能体开发实战:从零构建智能对话系统指南
java·开发语言·人工智能
折哥的程序人生 · 物流技术专研7 小时前
第8篇:六大设计原则在工厂模式中的体现
java·设计模式·设计原则·工厂模式·编程进阶·扩充系列
问君能有几多愁~7 小时前
C++ 数据结构复习笔记
数据结构·c++·笔记
至乐活着8 小时前
深入解析跳表SkipList:原理、实现与性能优化实战
数据结构·算法·跳表·skiplist·java实现
小明bishe188 小时前
计算机毕业设计之基于JAVA的植物科普网站
java·spring boot·spring·架构·课程设计
r_oo_ki_e_8 小时前
Java Map 集合学习笔记
java·笔记·学习
SL-staff8 小时前
智慧园区2000+设备统一管理:JVS-IoT如何降低运维成本40%
java·开发语言·物联网·智慧园区·设备管理·运维优化·jvs-iot
Jerry8 小时前
LeetCode 383. 赎金信
算法