原题如下
https://leetcode-cn.com/problems/matrix-block-sum/
题解
方法一
此方法无脑简单好想,就是在规定坐标范围内把所有数加起来,但要注意上下左右边界。
本思路java代码:
/*
*@v7fgg
*执行用时 :190 ms, 在所有 Java 提交中击败了19.68%的用户
*内存消耗 :40.3 MB, 在所有 Java 提交中击败了100.00%的用户
*2020年6月11日 20:12
*/
class Solution {
public int[][] matrixBlockSum(int[][] mat, int K) {
int m=mat.length;
int n=mat[0].length;
int ans[][]=new int[m][n];
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
int a=0;
for(int k=Math.max(0,i-K);k<=Math.min(i+K,m-1);k++){
for(int p=Math.max(j-K,0);p<=Math.min(j+K,n-1);p++){
a+=mat[k][p];
}
}
ans[i][j]=a;
}
}return ans;
}
}
方法二 前缀和
前缀和的意思就是,建立辅助阵,每个元素(i,j)表示的是矩阵mat在(i,j)范围内的数字和,该矩阵可有递推得到。计算本题的答案只需要对辅助矩阵里面的数字进行加减即可。
本思路java代码示例:
/*
*@v7fgg
*执行用时 :5 ms, 在所有 Java 提交中击败了88.68%的用户
*内存消耗 :40.7 MB, 在所有 Java 提交中击败了100.00%的用户
*2020年6月11日 21:19
*/
class Solution {
public int[][] matrixBlockSum(int[][] mat, int K) {
int m=mat.length;
int n=mat[0].length;
int ans[][]=new int[m][n];
int a[][]=new int[m][n];//前缀和
a[0][0]=mat[0][0];
for(int i=1;i<m;i++){
a[i][0]=a[i-1][0]+mat[i][0];
}
for(int j=1;j<n;j++){
a[0][j]=a[0][j-1]+mat[0][j];
}
for(int i=1;i<m;i++){
for(int j=1;j<n;j++){
a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+mat[i][j];
}
}
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
int m0=Math.max(0,i-K);
int n0=Math.max(0,j-K);
int m1=Math.min(m-1,i+K);
int n1=Math.min(n-1,j+K);
if(m0==0&&n0==0){
ans[i][j]=a[m1][n1];
}
else if(m0==0){
ans[i][j]=a[m1][n1]-a[m1][n0-1];
}
else if(n0==0){
ans[i][j]=a[m1][n1]-a[m0-1][n1];
}
else{
ans[i][j]=a[m1][n1]-a[m1][n0-1]-a[m0-1][n1]+a[m0-1][n0-1];
}
}
}
return ans;
}
}
本思路参考资料:https://leetcode-cn.com/problems/matrix-block-sum/solution/er-wei-shu-zu-qian-zhui-he-by-jin-ai-yi/