难度中等199
给定一个二维矩阵,计算其子矩形范围内元素的总和,该子矩阵的左上角为 (row1, col1)
,右下角为 (row2, col2)
。
上图子矩阵左上角 (row1, col1) = (2, 1) ,右下角(row2, col2) = (4, 3),该子矩形内元素的总和为 8。
思路:
采用303. 区域和检索 - 数组不可变的思路,利用在矩阵初始化过程中动态计算各位置中数据的统计情况。
根据统计信息计算对于的元素和仅需要简单的计算即可实现。
代码
class NumMatrix {
public:
vector<vector<int>> date;
bool flg=false;
int len_x=0,len_y=0;
NumMatrix(vector<vector<int>>& matrix) {
if(0==matrix.size()) { flg=true; return;}
len_x=matrix.size();
len_y=matrix[0].size();
date.resize(len_x + 1, vector<int>(len_y + 1)); // 更改尺寸时,全部覆盖为 零
// 初始化搜寻矩阵
for(int i=0;i<len_x;++i){
for(int j=0;j<len_y;++j){
date[i+1][j+1] = date[i+1][j]+date[i][j+1] + matrix[i][j] -date[i][j];
}
}
}
int sumRegion(int row1, int col1, int row2, int col2) {
if(flg ) return 0;
return date[row2+1][col2+1] - date[row2+1][col1] -date[row1][col2+1] + date[row1][col1];
}
};