leetcode1314.矩阵区域和

原题如下

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/