1314 矩阵区域和

题目描述：
给你一个 m * n 的矩阵 mat 和一个整数 K ，请你返回一个矩阵 answer ，其中每个 answer[i][j] 是所有满足下述条件的元素 mat[r][c] 的和：
i - K <= r <= i + K, j - K <= c <= j + K
(r, c) 在矩阵内。

示例 1：
输入：mat = [[1,2,3],[4,5,6],[7,8,9]], K = 1
输出：[[12,21,16],[27,45,33],[24,39,28]]

示例 2：
输入：mat = [[1,2,3],[4,5,6],[7,8,9]], K = 2
输出：[[45,45,45],[45,45,45],[45,45,45]]

提示：
m == mat.length
n == mat[i].length
1 <= m, n, K <= 100
1 <= mat[i][j] <= 100

方法1：
主要思路：解题链接汇总
（1）先计算原数组的二维前缀和；
（2）对原数组每个位置的有效范围，计算出该有效范围内的局部和；

class Solution {
public:
    vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int K) {
        vector<vector<int>> sum_sub(mat.size()+1,vector<int>(mat[0].size()+1,0));
        for(int i=1;i<=mat.size();++i){//计算前缀和
            for(int j=1;j<=mat[0].size();++j){
                sum_sub[i][j]=sum_sub[i-1][j]+sum_sub[i][j-1]-sum_sub[i-1][j-1]+mat[i-1][j-1];
            }
        }
        
        vector<vector<int>>res(mat.size(),vector<int>(mat[0].size()));
        for(int i=0;i<mat.size();++i){
            for(int j=0;j<mat[0].size();++j){
            	//获得当前位置的有效范围
                int r1=i<=K?0:i-K;
                int c1=j<=K?0:j-K;
                int r2=i+K<mat.size()?i+K:mat.size()-1;
                int c2=j+K<mat[0].size()?j+K:mat[0].size()-1;
                //求有效范围内的局部和
                res[i][j]=sum_sub[r2+1][c2+1]-sum_sub[r2+1][c1]-sum_sub[r1][c2+1]+sum_sub[r1][c1];
            }
        }
        return res;
    }
};