您现在的位置是:首页 >学无止境 >数组题目总结 -- 前缀和数组网站首页学无止境
数组题目总结 -- 前缀和数组
简介数组题目总结 -- 前缀和数组
目录
一. 区域和检索 - 数组不可变
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]。
二. 二维区域和检索 - 矩阵不可变
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/
风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。