304. Range Sum Query 2D

问题:

给定一个二维数组,

求给定区间围成矩形的和。

Example 1:
Input
["NumMatrix", "sumRegion", "sumRegion", "sumRegion"]
[[[[3, 0, 1, 4, 2], [5, 6, 3, 2, 1], [1, 2, 0, 1, 5], [4, 1, 0, 1, 7], [1, 0, 3, 0, 5]]], [2, 1, 4, 3], [1, 1, 2, 2], [1, 2, 2, 4]]
Output
[null, 8, 11, 12]

Explanation
NumMatrix numMatrix = new NumMatrix([[3, 0, 1, 4, 2], [5, 6, 3, 2, 1], [1, 2, 0, 1, 5], [4, 1, 0, 1, 7], [1, 0, 3, 0, 5]]);
numMatrix.sumRegion(2, 1, 4, 3); // return 8 (i.e sum of the red rectangele).
numMatrix.sumRegion(1, 1, 2, 2); // return 11 (i.e sum of the green rectangele).
numMatrix.sumRegion(1, 2, 2, 4); // return 12 (i.e sum of the blue rectangele).
 

Constraints:

m == matrix.length
n == matrix[i].length
1 <= m, n <= 200
-10^5 <= matrix[i][j] <= 10^5
0 <= row1 <= row2 < m
0 <= col1 <= col2 < n
At most 10^4 calls will be made to sumRegion.

  

example 1:

解法:DP,preSum

1. 状态:dp[i][j]:[0,0]~[i-1,j-1]的矩形和

  • i:i坐标
  • j:j坐标

2.选择:

  • dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]

3.base:

  • dp[0][j]=0
  • dp[i][0]=0

而题目所求矩形[i1,j1]~[i2,j2]的和为:

  • res=dp[i2][j2]-dp[i2][j1]-dp[i1][j2]+dp[i1][j1]

代码参考:

 1 class NumMatrix {
 2 public:
 3     vector<vector<int>> dp;
 4     NumMatrix(vector<vector<int>>& matrix) {
 5         int n = matrix.size();
 6         int m = matrix[0].size();
 7         dp.resize(n+1, vector<int>(m+1, 0));
 8         for(int i=1; i<=n; i++) {
 9             for(int j=1; j<=m; j++) {
10                 dp[i][j]=dp[i][j-1]+dp[i-1][j]-dp[i-1][j-1]+matrix[i-1][j-1];
11             }
12         }
13         return;
14     }
15     
16     int sumRegion(int row1, int col1, int row2, int col2) {
17         return dp[row2+1][col2+1]-dp[row2+1][col1]-dp[row1][col2+1]+dp[row1][col1];
18     }
19 };
20 
21 /**
22  * Your NumMatrix object will be instantiated and called as such:
23  * NumMatrix* obj = new NumMatrix(matrix);
24  * int param_1 = obj->sumRegion(row1,col1,row2,col2);
25  */
原文地址:https://www.cnblogs.com/habibah-chang/p/14622038.html