首页 > 编程语言 >leetcode——数组算法——前缀和构建和应用

leetcode——数组算法——前缀和构建和应用

时间:2024-02-13 23:24:27浏览次数:39  
标签:前缀 nums int preMatrix 算法 sumRange leetcode left

leetcode——数组算法——前缀和构建和应用

前缀和技巧适用于快速、频繁地计算一个索引区间内的元素之和

303. 区域和检索 - 数组不可变

比如leetcode 303. 区域和(检索 - 数组不可变)

题目介绍:

给定一个整数数组 nums,处理以下类型的多个查询:

  1. 计算索引 leftright (包含 leftright)之间的 nums 元素的 ,其中 left <= right

实现 NumArray 类:

  • NumArray(int[] nums) 使用数组 nums 初始化对象
  • int sumRange(int i, int j) 返回数组 nums 中索引 leftright 之间的元素的 总和 ,包含 leftright 两点(也就是 nums[left] + nums[left + 1] + ... + nums[right] )

示例 1:

输入:
["NumArray", "sumRange", "sumRange", "sumRange"]
[[[-2, 0, 3, -5, 2, -1]], [0, 2], [2, 5], [0, 5]]
输出:
[null, 1, -1, -3]

解释:
NumArray numArray = new NumArray([-2, 0, 3, -5, 2, -1]);
numArray.sumRange(0, 2); // return 1 ((-2) + 0 + 3)
numArray.sumRange(2, 5); // return -1 (3 + (-5) + 2 + (-1)) 
numArray.sumRange(0, 5); // return -3 ((-2) + 0 + 3 + (-5) + 2 + (-1))

解法一:

1.在sumRange里面,for循环从left到right遍历nums,用一个变量记录。

代码如下:

class NumArray {
    //类里肯定有一个int[]成员
    private int[] myArray;
    public NumArray(int[] nums) {
        this.myArray=nums;
    }
    
    public int sumRange(int left, int right) {
        int result=0;
        if(left>right||left<0||right>myArray.length){
            return 0;
        }
        //myArray[left]一直加到myArray[right]
        for(int i=left;i<=right;i++){
            result+=myArray[i];
        }
        return result;
    }
}

如果多次调用sumRange,会一直重复计算。

解法2

2.在构造函数中,构造一个关于nums的前缀和数组preNums,preNums[i]的值就是nums前i项的和。

Q:如何构造这个前缀和数组?

A:前缀和数组的每一项 = 前一项(前i-1项的和)+ nums[i]。

注意:因为前缀和数组的表达意义应该是前1项的和,前2项的和;而没有个前0项的和。

所以这里将preNum[0]=0;目的是更符合我们的表达语义。

比如preNum[1]就是nums前1项的和。

代码如下:

class NumArray {
    public int[] getPreArray() {
        return preArray;
    }

    //记录一个前缀和数组,避免sumRange重复的for
    private int[] preArray;
    public NumArray(int[] nums) {
        preArray = new int[nums.length + 1];
        // 计算 nums 的累加和
        for (int i = 1; i < preArray.length; i++) {
            preArray[i] = preArray[i - 1] + nums[i - 1];
        }
    }

    public int sumRange(int left, int right) {
        int result=preArray[right+1]-preArray[left];
        return result;
    }
}

304. 二维区域和检索 - 矩阵不可变

如果是二维数组的前缀和如何构建和使用呢?

比如leetcode 304. 二维区域和检索 - 矩阵不可变

给定一个二维矩阵 matrix,以下类型的多个请求:

  • 计算其子矩形范围内元素的总和,该子矩阵的 左上角(row1, col1)右下角(row2, col2)

实现 NumMatrix 类:

  • NumMatrix(int[][] matrix) 给定整数矩阵 matrix 进行初始化
  • int sumRegion(int row1, int col1, int row2, int col2) 返回 左上角 (row1, col1)右下角 (row2, col2) 所描述的子矩阵的元素 总和

示例 1:

输入: 
["NumMatrix","sumRegion","sumRegion","sumRegion"]
[[[[3,0,1,4,2],[5,6,3,2,1],[1,2,0,1,5],[4,1,0,1,7],[1,0,3,0,5]]],[2,1,4,3],[1,1,2,2],[1,2,2,4]]
输出: 
[null, 8, 11, 12]

解释:
NumMatrix numMatrix = new NumMatrix([[3,0,1,4,2],[5,6,3,2,1],[1,2,0,1,5],[4,1,0,1,7],[1,0,3,0,5]]);
numMatrix.sumRegion(2, 1, 4, 3); // return 8 (红色矩形框的元素总和)
numMatrix.sumRegion(1, 1, 2, 2); // return 11 (绿色矩形框的元素总和)
numMatrix.sumRegion(1, 2, 2, 4); // return 12 (蓝色矩形框的元素总和)

如果本题继续双for循环,开销很大,如果sumRegion使用频繁,则可以使用一个前缀和数组存储NumMatrix前i行前j列的和。

核心

Q:二维数组的前缀和如何构建呢?

A:行列的length各+1,然后找规律:左面的+上面的+自己-左对角线的

Q:规律怎么找的?

A:比如上图中的matrix(2)(2),它值为0;现在要计算前3行前3列的前缀和。

注意它左边的2和上面的3,如果让他俩各自位置的前缀和相加,然后再减去对角线的6位置的前缀和,就是0位置的前缀和。

如下图所示(可以好好屡屡):

代码如下:

class NumMatrix {
    public int[][] getPreMatrix() {
        return preMatrix;
    }

    private int[][] preMatrix;
    public NumMatrix(int[][] matrix) {
        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][j-1]+preMatrix[i-1][j]+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];
    }

    public static void main(String[] args) {
        int[][] matrix = {
                {3, 0, 1, 4, 2},
                {5, 6, 3, 2, 1},
                {1, 2, 0, 1, 5},
                {4, 1, 0, 1, 7},
                {1, 0, 3, 0, 5}
        };

        NumMatrix numMatrix = new NumMatrix(matrix);
        System.out.println(Arrays.deepToString(numMatrix.getPreMatrix()));
        System.out.println(numMatrix.sumRegion(2, 1, 4, 3));

    }
}

标签:前缀,nums,int,preMatrix,算法,sumRange,leetcode,left
From: https://www.cnblogs.com/xsr2004/p/18014904

相关文章

  • 【算法】DFS
    DFSDFS是一种遍历或搜索图、树或图像等数据结构的算法,当然这个图、树未必存储下来。常见的是通过某种关系构造出的搜索树,搜索树一般是排列型搜索树和子集型搜索树。DFS一般使用栈或递归。一道模板题#include<bits/stdc++.h>usingnamespacestd;constintN=50;intn,m......
  • 基于NIQE算法的图像无参考质量评价算法matlab仿真
    1.算法运行效果图预览   2.算法运行软件版本MATLAB2022a  3.算法理论概述      NIQE(NaturalnessImageQualityEvaluator)算法是一种无参考图像质量评价算法,旨在评估图像的自然度,即图像看起来是否像自然场景。NIQE基于一组“质量感知”特征,并将其拟合到MV......
  • OpenCL规约算法例子
    本文给出一个规约算法求数组的和的例子。本例子求20000000(两千万)个整数的和。运算过程分成了两步,第一步是GPU对每一个工作组内规约求和,然后将每个工作组的求和结果放到数组中输出。第二步是对输出的数组用CPU求和。实际运行对比发现GPU的效率不如用CPU直接求和。下述算法运行环境......
  • 2024牛客寒假算法基础集训营3
    M题智乃的36倍数(normalversion)错解幂运算写成了乘~#include<bits/stdc++.h>usingnamespacestd;#defineendl'\n'#defineintlonglong#definedebug(x)cout<<x<<""<<endl;#define_debug(a,n)for(inti=0;i<n;i++)......
  • [LeetCode] 2108. Find First Palindromic String in the Array
    Givenanarrayofstringswords,returnthefirstpalindromicstringinthearray.Ifthereisnosuchstring,returnanemptystring"".Astringispalindromicifitreadsthesameforwardandbackward.Example1:Input:words=["abc&quo......
  • 2024牛客寒假算法基础集训营1
    2024牛客寒假算法基础集训营1A解题思路:按照\(dfs\)出现顺序暴力判断即可。代码:#include<bits/stdc++.h>usingnamespacestd;usingll=longlong;usingpii=pair<ll,ll>;#definefifirst#definesesecondusingi128=__int128_t;usingpiii=pair<ll,pai......
  • boruvka 算法学习笔记
    boruvka算法就是最小生成树B算法。B算法的思路是每次对每个连通块,求出它能连出去的权值最小的边,然后再按边权从小到大合并。由于每次操作连通块数至少减半,所以复杂度是\(O(m\logn)\)。1.CF1305GKuroniandAntihype题意:长为\(n\)的数列\(a\),现在要选择全部数,每一次你......
  • 微分积分及其运算法则成对列举
         ......
  • 【算法】排序
    冒泡排序思想:每一轮确定一个最大值移到最右边。点击查看代码for(inti=n;i>=1;i--){ for(intj=1;j<i;j++){ if(a[j]>a[j+1])swap(a[j],a[j+1]); } }选择排序思想:与冒泡相似。点击查看代码for(inti=n;i>=1;i--){ intmax_id......
  • 2024牛客寒假算法基础集训营2个人补题题解(K、D)
    比赛链接:2024牛客寒假算法基础集训营2K、TokitsukazeandPassword(easy)题面看着很难实际上只要暴力的东西,赛时看了眼题面就溜了血亏爆搜,枚举\(abcd\)和_可能的值,枚举的情况只有\(9*8*7*6*9=27216\)种。判断按照枚举出的对应值排列出的密码是否满足条件,满足就\(ans++\)写完......