目录
- 一. 区域和检索 - 数组不可变
- 1. 思路和代码
- I. 博主的做法：
- II. 东哥的做法：
- 2. 总结
- 二. 二维区域和检索 - 矩阵不可变
- 1. 思路和代码
- I. 博主的做法：
- II. 东哥的做法：
- 2. 总结
一. 区域和检索 - 数组不可变
- 题目链接：https://leetcode.cn/problems/range-sum-query-immutable/
1. 思路和代码
I. 博主的做法：
- 很简单，先在类里面整一个private数组，然后初始化它，只有就是简单的遍历求和。
class NumArray {
private int[] nummber;
public NumArray(int[] nums) {
this.nummber = nums;
}
public int sumRange(int left, int right) {
int sum = 0;
for(int i = left; i <= right; i++)
sum += nummber[i];
return sum;
}
}
II. 东哥的做法：
- 博主这样写，可以达到效果，但是效率很差，因为 sumRange 方法会被频繁调用，而它的时间复杂度是 O(N)，其中 N 代表 nums 数组的长度。
- 构建前缀和数组preSum，这样，sumRange 函数仅仅需要做一次减法运算，避免了每次进行 for 循环调用，最坏时间复杂度为常数 O(1)。
class NumArray {
private int[] preSum;
public NumArray(int[] nums) {
preSum = new int[nums.length + 1];
for(int i = 1; i < preSum.length; i++)
preSum[i] = preSum[i-1] + nums[i-1];
}
public int sumRange(int left, int right) {
return preSum[right + 1] - preSum[left];
// return preSum[0];
}
}
-
- 这个技巧在生活中运用也挺广泛的，比方说，你们班上有若干同学，每个同学有一个期末考试的成绩（满分 100 分），那么请你实现一个 API，输入任意一个分数段，返回有多少同学的成绩在这个分数段内。那么，你可以先通过计数排序的方式计算每个分数具体有多少个同学，然后利用前缀和技巧来实现分数段查询的 API：
package LeetCode;
import java.util.*;
/**
* @Author wangyichao
* @date 2022/3/22 12:59
* @Version 1.0
* @Description
*/
public class Test {
//前缀和数组
private int[] preCount;
public Test(int[] score){
preCount = new int[score.length + 1];
//初始化前缀和数组
for(int i = 1; i < preCount.length; i++)
preCount[i] = preCount[i-1] + score[i-1];
}
//统计分数段[left_core , right_core]中学生的个数
public int sum(int left_core, int right_core){
return preCount[right_core + 1] - preCount[left_core];
}
//使用计数排序(桶排序)，统计每一个分数的学生个数
public static int[] add(int[] score){
int[] num = new int[100 + 1];
for(int score1 : score)
num[score1]++;
return num;
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int left = sc.nextInt();
int right = sc.nextInt();
int[] score = add(new int[]{0,1,1,2,3,3,3,4,5});
Test test = new Test(score);
System.out.println(test.sum(left,right));
}
}
2. 总结
- 博主本来想让 preSum[0] 来存储 sum 的值，最后发现结果不对，因为 preSum[0] 这个值也要参与到求和的运算当中。eg：求下标 0 ~ 下标 5 之间元素的和，那么就是 preSum[5 + 1] - preSum[0]。
二. 二维区域和检索 - 矩阵不可变
- 题目链接：https://leetcode.cn/problems/range-sum-query-2d-immutable/
1. 思路和代码
I. 博主的做法：
- 博主就是将二维数组的变成很多个一维数组，用多个一维数组构建前缀和数组。
class NumMatrix {
private int[][] preMatrix;
public NumMatrix(int[][] matrix) {
if(matrix.length == 0 || matrix[0].length == 0)
return;
preMatrix = new int[matrix.length][matrix[0].length + 1];
for(int i = 0; i < matrix.length; i++){
for(int j = 1; j < preMatrix[0].length; j++){
preMatrix[i][j] = preMatrix[i][j-1] + matrix[i][j-1];
}
}
}
public int sumRegion(int row1, int col1, int row2, int col2) {
int sum = 0;
for(int i = row1; i <= row2; i++)
sum += preMatrix[i][col2 + 1] - preMatrix[i][col1];
return sum;
}
}
II. 东哥的做法：
- 类似于前缀和的做法， 先算出（0，0，row + 1，col + 1）这个大矩阵的和，也就是下图中绿色方框中的和，然后再减去剩下两个矩形的和，中间有一部分是多减的，现在把它加回来，最后得到了红色矩形的和，也就是我们想得到的结果。
- 我们可以维护一个二维 preMatrix 数组，专门记录以原点为顶点的矩阵的元素之和，就可以用几次加减运算算出任何一个子矩阵的元素和。
- 这样，sumRegion 函数的时间复杂度也用前缀和技巧优化到了 O(1)，这是典型的「空间换时间」思路。
class NumMatrix {
private int[][] preMatrix;
public NumMatrix(int[][] matrix) {
if(matrix.length == 0 || matrix[0].length == 0)
return;
preMatrix = new int[matrix.length + 1][matrix[0].length + 1];
for(int i = 1; i < preMatrix.length; i++){
for(int j = 1; j < preMatrix[0].length; j++){
preMatrix[i][j] = preMatrix[i-1][j] + preMatrix[i][j-1] + matrix[i-1][j-1] - preMatrix[i-1][j-1];
}
}
}
public int sumRegion(int row1, int col1, int row2, int col2) {
return preMatrix[row2+1][col2+1] - preMatrix[row1][col2+1] - preMatrix[row2+1][col1] + preMatrix[row1][col1];
}
}
这里preMatrix的构成还有最后的sumRegion都很迷，博主举个例子帮助大家理解：
score为输入的数组，preMatrix为前缀和数组。
- 以preMatrix当中的14为例：
preMatrix[2][2] = preMatrix[1][2] + preMatrix[2][1] + matrix[1][1] - preMatrix[1][1]
14 (3 + 5 + 6) = 3 (0 + 3) + 8 (3 + 5) + 6 - 3
对应的Matrix如下图：
- 假设要算(1, 1, 2, 2)这个矩阵（黑色方框中矩形）的和：
sum = preMatrix[3][3] - preMatrix[1][3] - preMatrix[3][1] + preMatrix[1][1]
11 = 21 (3 + 0 + 1 + 5 + 6 + 3 + 1 + 2 + 0) - 4 (3 + 0 + 1)- 9（3 + 5 + 1） + 3
- 本质上还是矩阵的相互加减
2. 总结
- 博主的那个方法，虽然是前缀数组的衍生，但sumRegion时间复杂度为O(n)，而东哥的这个时间复杂度为O(1)。
- 很巧妙，运用迭代的思想构建preMartix数组，又运送矩阵相减的方法，算出sum，东哥牛逼！！！！
-二维数组.length 返回的是二维数组行的个数，而列的个数要用二维数组[0].length来表示。 - 一维数组中，只有第一个元素是空的；二维数组中，第一行第一列都是空的。发现一个规律：preMatrix的行数列数都减一就是matrix中对应的矩阵(0, 0, row, col)右下角点的坐标
- eg：以上面那个例子接着来说，preMatrix[3][3]： (2 ,2)即是大矩阵右下角的坐标，那么整个大矩阵就是（0, 0, 2, 2）,也就是下面这个图
- eg：以上面那个例子接着来说，preMatrix[3][3]： (2 ,2)即是大矩阵右下角的坐标，那么整个大矩阵就是（0, 0, 2, 2）,也就是下面这个图
参考：https://labuladong.github.io/algo/di-yi-zhan-da78c/shou-ba-sh-48c1d/xiao-er-me-03265/
