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

原创已于 2022-04-08 23:34:50 修改 · 404 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #c++ #算法

于 2022-04-08 23:33:47 首次发布

leetcode 专栏收录该内容

79 篇文章

订阅专栏

该博客介绍了如何使用C++实现二维前缀和数据结构，以高效计算矩形子矩阵的和。NumMatrix类通过预处理矩阵的每个元素，实现了在常数时间内查询指定区域的元素总和。方法包括累加上下左右相邻元素，以及修正左上角对角线的值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

在这里插入图片描述

结果

//题目类型：二维前缀和
class NumMatrix {
public:
    vector<vector<int>> ans;
    int n, m;
    NumMatrix(vector<vector<int>>& matrix) {
        n = matrix.size(), m = matrix[0].size();
        ans.resize(n, vector<int>(m, 0));
        for (int i = 0; i < n; ++i)
            for (int j = 0; j < m; ++j) {
                ans[i][j] = matrix[i][j];
                i && (ans[i][j] += ans[i - 1][j]); //加上正上方位置的值
                j && (ans[i][j] += ans[i][j - 1]); //加上左边位置的值
                i && j && (ans[i][j] -= ans[i - 1][j - 1]); //减去左上角对角线的值               
            }
    }
    
    int sumRegion(int row1, int col1, int row2, int col2) {
        int t = ans[row2][col2];
        row1 && (t -= ans[row1 - 1][col2]); //减去子矩阵上方
        col1 && (t -= ans[row2][col1 - 1]); //减去子矩阵左侧
        row1 && col1 && (t += ans[row1 - 1][col1 - 1]); //加上子矩阵左上角
        return t;
    }
};

/**
 * Your NumMatrix object will be instantiated and called as such:
 * NumMatrix* obj = new NumMatrix(matrix);
 * int param_1 = obj->sumRegion(row1,col1,row2,col2);
 */