前缀和技巧适用于快速、频繁地计算一个索引区间内的元素之和。
一维数组的前缀和
若求区域和的方法sumRange ()存在多次调用, 那么在求子区域元素之和时,每次都逐个相加计算并不是理想的做法。
可以维护一个一维数组preSum,preSum[i]存储nums的前i个元素的和,则可得到[i, j]子区间的和为preSum[j]-preSum[i-1]。
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]。
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];
}
};