C++의 행렬 블록 합

mat라고 하는 하나의 m * n 행렬과 정수 K가 있다고 가정하면 각 대답[i][j]이 모든 요소 mat[ r][c] for i - K <=r <=i + K, j - K <=c <=j + K, 그리고 (r, c)는 행렬의 유효한 위치입니다. 따라서 입력이 다음과 같으면 -

k가 1이면 출력은 -

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

• n :=행 수, m =열 수
• n x m의 순서를 갖는 행렬 ans 정의
• 0 ~ n – 1 범위의 i에 대해
  • 0 ~ m – 1 범위의 j에 대해
    • i – k ~ i + k 범위의 r에 대해
      • j – k ~ j + k 범위의 c
        • r과 c가 행렬 인덱스 안에 있으면
          • ans[i, j] :=ans[i, j] + mat[r, c]
• 반환

예(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, ],]