Leetcode 区域和检索

最新推荐文章于 2024-04-12 17:50:38 发布

打盹码农

最新推荐文章于 2024-04-12 17:50:38 发布

阅读量127

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/AHASAN/article/details/116018752

版权

LeetCode 专栏收录该内容

51 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用前缀和技巧解决数组和矩阵的求和问题。文中给出了两个示例，分别是单维数组的区域和检索以及二维矩阵的子矩形和检索，通过构建前缀和数组来实现高效查询。这两个数据结构在不改变原始数组或矩阵的情况下，能够快速计算指定范围内的元素总和。

摘要由CSDN通过智能技术生成

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

给定一个整数数组 nums，求出数组从索引 i 到 j（i ≤ j）范围内元素的总和，包含 i、j 两点。

class NumArray {
    int[] sums;
    public NumArray(int[] nums) {
        sums=new int[nums.length+1];//前缀和数组长度设置为数组长度+1，避免判断left=0的情况
        sums[0]=0;//left=0时，right前缀和只需要-0
        for(int i=0;i<nums.length;i++){
            sums[i+1]=sums[i]+nums[i];
        }
    }
    
    public int sumRange(int left, int right) {
        return sums[right+1]-sums[left];
    }
}

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

给定一个二维矩阵，计算其子矩形范围内元素的总和，该子矩阵的左上角为 (row1, col1) ，右下角为 (row2, col2) 。

上图子矩阵左上角 (row1, col1) = (2, 1) ，右下角(row2, col2) = (4, 3)，该子矩形内元素的总和为 8。

示例：
给定 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]
]

sumRegion(2, 1, 4, 3) -> 8
sumRegion(1, 1, 2, 2) -> 11
sumRegion(1, 2, 2, 4) -> 12
提示：

你可以假设矩阵不可变。
会多次调用 sumRegion 方法。
你可以假设 row1 ≤ row2 且 col1 ≤ col2 。

class NumMatrix {
    int[][] preSum;
    public NumMatrix(int[][] matrix) {
        int m=matrix.length;
        int n=matrix[0].length;
        preSum=new int[m+1][n+1];//二维前缀和
        for(int i=1;i<=m;i++){
            for(int j=1;j<=n;j++){
                preSum[i][j]=preSum[i-1][j]+preSum[i][j-1]-preSum[i-1][j-1]+matrix[i-1][j-1];
            }
        }
    }
    public int sumRegion(int row1, int col1, int row2, int col2) {
        return preSum[row2+1][col2+1]+preSum[row1][col1]-preSum[row2+1][col1]-preSum[row1][col2+1];
    }
}