Range Sum Query 2D - Immutable

本文介绍了一个使用 C++ 实现的 NumMatrix 类,该类能够高效地对矩阵区域求和。构造函数接受一个二维整数矩阵,并通过预处理计算累积和来加速后续的区域求和操作。sumRegion 方法返回指定矩形区域内的元素之和。

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

c++

class NumMatrix {
public:
    NumMatrix(vector<vector<int>> &matrix) {
        if (matrix.empty())
            return;
        cum = matrix;
        for (int i = 0; i < cum.size(); ++i) {
            for (int j = 0; j < cum[i].size(); ++j) {
                if (i == 0 && j == 0) {
                    cum[i][j] = cum[i][j]; continue;
                }
                if (i == 0) {
                    cum[i][j] = cum[i][j] + cum[i][j - 1]; continue;
                } 
                if (j == 0) {
                    cum[i][j] = cum[i][j] + cum[i - 1][j]; continue;
                }
                cum[i][j] = cum[i][j] + cum[i][j - 1] + cum[i - 1][j] - cum[i - 1][j - 1];
            }
        }
    }

    int sumRegion(int row1, int col1, int row2, int col2) {
        if (cum.empty()) return 0;
        if (row1 == 0 && col1 == 0) return cum[row2][col2];
        if (row1 == 0) return cum[row2][col2] - cum[row2][col1 - 1];
        if (col1 == 0) return cum[row2][col2] - cum[row1 - 1][col2];
        return cum[row2][col2] - cum[row2][col1 - 1] - cum[row1 - 1][col2] + cum[row1 - 1][col1 - 1];
    }
private:
    vector<vector<int>> cum;
};

reference:
http://blog.sina.com.cn/s/blog_4cb0b54301017wwo.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值