304. Range Sum Query 2D - Immutable

最后更新

四刷

10-Jan-2017

白送的题。

注意3点：

1) int[][] dp最好长宽都多加一行。要不累加的时候第一行第一列都要当做特殊情况处理= =

2) 注意index的选取= =容易混乱

3) 我他妈了个逼的长得太帅了。

Time:

init: O(mn)

query: O(1)

Space: O(mn)

public class NumMatrix {
int[][] dp;

public NumMatrix(int[][] matrix) {
if (matrix.length == 0) return;
dp = new int[matrix.length + 1][matrix[0].length + 1];
for (int i = 1; i < dp.length; i++) {
for (int j = 1; j < dp[0].length; j++) {
dp[i][j] += matrix[i-1][j-1] + dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1];
}
}
}

public int sumRegion(int row1, int col1, int row2, int col2) {
return dp[row2 + 1][col2 + 1] + dp[row1][col1]
-dp[row1][col2 + 1] - dp[row2 + 1][col1];
}
}

二刷。

思路对了，不过做起来麻烦。主要是构建2-D dp Array的时候没有在原先基础上多加1行1列，这样各种EDGE CAES必须判断一下，否则出界。。

多加一行一列就没这么多事了。

dp[i][j]是(i,j)为右下角的矩形（的面积）。

求某个矩形，用dp[i][j] - dp[i][j-1] - dp[i-1][j] + dp[i-1][j-1]就行了，但是我们定义的时候多定义了一行，所以对应坐标要跟着改变1。

public class NumMatrix {

int[][] dp;
int shit;
public NumMatrix(int[][] matrix)
{
if(matrix.length == 0)
{
this.shit = 0;

}
else
{
this.dp = new int[matrix.length+1][matrix[0].length+1];
this.shit = 1;

for(int i = 1; i <= matrix.length;i++)
{
for(int j = 1; j <= matrix[0].length;j++)
{
dp[i][j] = matrix[i-1][j-1] + dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1];
}
}

}

}

public int sumRegion(int row1, int col1, int row2, int col2)
{
if(this.shit == 0) return 0;
return dp[row2+1][col2+1] - dp[row2+1][col1] - dp[row1][col2+1] + dp[row1][col1];
}
}

三刷。

白送的。。dp[][]

4个正方形D-B-C+A

public class NumMatrix {
int[][] dp;
int m;
int n;
public NumMatrix(int[][] matrix) {
if (matrix.length == 0) return;
m = matrix.length;
n = matrix[0].length;
dp = new int[m+1][n+1];
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
dp[i+1][j+1] = dp[i+1][j] + dp[i][j+1] - dp[i][j] + matrix[i][j];
}
}
}

public int sumRegion(int row1, int col1, int row2, int col2) {
return dp[row2+1][col2+1] - dp[row2+1][col1] - dp[row1][col2+1] + dp[row1][col1];
}
}