[二维前缀和]304. 二维区域和检索 - 矩阵不可变

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

304. 二维区域和检索 - 矩阵不可变 - 力扣(LeetCode)

Leetcode-304.二维区域和检索-矩阵不可变_哔哩哔哩_bilibili

class NumMatrix:

    def __init__(self, matrix: List[List[int]]):
        m, n = len(matrix),len(matrix[0])
        s = [[0]*(n+1) for _ in range(m+1)]
        for i,row in enumerate(matrix):
            for j,x in enumerate(row):
                s[i+1][j+1] = s[i+1][j] + s[i][j+1] -s[i][j]+x
        self.s = s

    def sumRegion(self, row1: int, col1: int, row2: int, col2: int) -> int:
        return self.s[row2+1][col2+1] - self.s[row2+1][col1]-self.s[row1][col2+1]+self.s[row1][col1]

        


# Your NumMatrix object will be instantiated and called as such:
# obj = NumMatrix(matrix)
# param_1 = obj.sumRegion(row1,col1,row2,col2)

二维前缀和是求二维数组上从(0,0)位置到(i,j)位置的累加,其构建方法使用方式如下: ### 构建方法 以模板“二维区域检索 - 矩阵不可变”为例,假设有一个二维矩阵`matrix`,构建二维前缀和数组`sum`的步骤如下: 1. 获取矩阵的行数`n`列数`m`,并初始化前缀和数组`sum`的大小为`(n + 2) x (m + 2)`,初始值都为 0。 2.矩阵`matrix`的值赋给前缀和数组`sum`,注意索引的转换,即`sum[i][j] = matrix[i - 1][j - 1]`。 3. 计算二维前缀和,公式为`sum[i][j] += sum[i][j - 1] + sum[i - 1][j] - sum[i - 1][j - 1]`。 以下是对应的 C++ 代码实现: ```cpp #include <vector> using namespace std; class NumMatrix { public: vector<vector<int>> sum; NumMatrix(vector<vector<int>>& matrix) { int n = matrix.size(); int m = matrix[0].size(); sum.resize(n + 2, vector<int>(m + 2, 0)); for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { sum[i][j] = matrix[i - 1][j - 1]; } } // 二维前缀和 for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { sum[i][j] += sum[i][j - 1] + sum[i - 1][j] - sum[i - 1][j - 1]; } } } }; ``` ### 使用方式 二维前缀和主要用于快速计算二维数组中任意子矩阵。若要计算从`(row1, col1)`到`(row2, col2)`的子矩阵,可以使用以下公式: `sumRegion = sum[row2][col2] - sum[row2][col1 - 1] - sum[row1 - 1][col2] + sum[row1 - 1][col1 - 1]` 以下是使用上述公式计算子矩阵的 C++ 代码: ```cpp int sumRegion(int row1, int col1, int row2, int col2) { row1++; row2++; col1++; col2++; return sum[row2][col2] - sum[row2][col1 - 1] - sum[row1 - 1][col2] + sum[row1 - 1][col1 - 1]; } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值