一维矩阵和二维矩阵的前缀和

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

一维数组的前缀和

若求区域和的方法sumRange ()存在多次调用, 那么在求子区域元素之和时,每次都逐个相加计算并不是理想的做法。
可以维护一个一维数组preSum,preSum[i]存储nums的前i个元素的和,则可得到[i, j]子区间的和为preSum[j]-preSum[i-1]

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

class NumArray { //前缀和
    vector<int> presum;
public:
    NumArray(vector<int>& nums) {
        presum.push_back(0);
        for(int i=0;i<nums.size();i++)
            presum.push_back(presum.back()+nums[i]);
    }
    
    int sumRange(int left, int right) {
        return presum[right+1]-presum[left];
    }
}

二维矩阵的前缀和

可以维护一个二维数组 preSum ,记录以原点为顶点的矩阵的元素之和,就可以用几次加减运算算出任何一个子矩阵的元素和。
preSum外围初始化一圈0值,方便计算。

如下图所示,橙框 = 黑框 + 绿框 - 粉框 + matrix[i-1][j-1] (注意preSum的下标比matrix下标大1),即 preSum[i][j] = preSum[i][j-1] + preSum[i-1][j] - preSum[i-1][j-1] + matrix[i-1][j-1]。

在这里插入图片描述

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

class NumMatrix {
    vector<vector<int>> preSum;
public:
    NumMatrix(vector<vector<int>>& matrix) {
        preSum.resize(matrix.size()+1);
        for(int i=0;i<=matrix.size();i++){
            preSum[i].resize(matrix[0].size()+1);
            if(!i) continue; //第0行保留为全0
            for(int j=1;j<=matrix[0].size();j++){
                preSum[i][j]=preSum[i][j-1]+preSum[i-1][j]-preSum[i-1][j-1]+matrix[i-1][j-1];
            }
        }
    }
    
    int sumRegion(int row1, int col1, int row2, int col2) {
        return preSum[row2+1][col2+1]-preSum[row1][col2+1]-preSum[row2+1][col1]+preSum[row1][col1];
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值