[LeetCode] Range Sum Query 2D - Immutable

Very similar to Range Sum Query - Immutable, but we now need to compute a 2d accunulated-sum. In fact, if you work in computer vision, you may know a name for such an array --- Integral Image.

To solve this problem, Stefan has already posted a very elegant solution.

The code is copied here.

 1 class NumMatrix {
 2 public:
 3     NumMatrix(vector<vector<int>> &matrix) {
 4         accu = matrix;
 5         for (int i = 0; i < matrix.size(); i++)
 6             for (int j = 0; j < matrix[0].size(); j++)
 7                 accu[i][j] += a(i-1, j) + a(i, j-1) - a(i-1, j-1);
 8     }
 9 
10     int sumRegion(int row1, int col1, int row2, int col2) {
11         return a(row2, col2) - a(row1-1, col2) - a(row2, col1-1) + a(row1-1, col1-1);
12     }
13 private:
14     vector<vector<int>> accu;
15     int a(int i, int j) {
16         return i >= 0 && j >= 0 ? accu[i][j] : 0;
17     }
18 };
19 
20 
21 // Your NumMatrix object will be instantiated and called as such:
22 // NumMatrix numMatrix(matrix);
23 // numMatrix.sumRegion(0, 1, 2, 3);
24 // numMatrix.sumRegion(1, 2, 3, 4);

 

posted @ 2015-11-12 22:00  jianchao-li  阅读(666)  评论(0编辑  收藏  举报