C ++中的矩阵块总和
假设我们有一个名为mat的m*n矩阵和一个整数K，我们必须找到另一个矩阵答案，其中每个答案[i][j]是i-K<=的所有元素mat[r][c]的总和r<=i+K，j-K<=c<=j+K，并且(r,c)是矩阵中的有效位置。所以如果输入像-
且k为1，则输出为-
为了解决这个问题，我们将遵循以下步骤-
n：=行数，m=列数
定义一个矩阵ans，其顺序为nxm
对于i，范围为0至n–1
对于在i–k到i+k范围内的r
如果r和c在矩阵索引内，则
ans[i，j]：=ans[i，j]+mat[r，c]
对于范围j–k至j+k的c
对于j，范围从0到m–1
返回ans
例子（C++）
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<vector<auto> > v){
cout << "[";
for(int i = 0; i<v.size(); i++){
cout << "[";
for(int j = 0; j <v[i].size(); j++){
cout << v[i][j] << ", ";
}
cout << "],";
}
cout << "]"<<endl;
}
class Solution {
public:
vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int k) {
int n = mat.size();
int m = mat[0].size();
vector < vector <int> > ans(n , vector <int> (m));
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
for(int r = i - k;r <= i + k; r++){
for(int c = j - k; c <= j + k; c++){
if(r>= 0 && r < n && c >= 0 && c < m){
ans[i][j] += mat[r][c];
}
}
}
}
}
return ans;
}
};
main(){
vector<vector<int>> v1 = {{1,2,3},{4,5,6},{7,8,9}};
Solution ob;
print_vector(ob.matrixBlockSum(v1, 1));
}输入值
[[1,2,3],[4,5,6],[7,8,9]] 1
输出结果
[[12, 21, 16, ],[27, 45, 33, ],[24, 39, 28, ],]