C++ 中的矩陣塊和

假設我們有一個 m * n 矩陣 mat 和一個整數 K，我們需要找到另一個矩陣 answer，其中每個 answer[i][j] 是所有元素 mat[r][c] 的總和，對於 i - K <= r <= i + K，j - K <= c <= j + K，並且 (r, c) 是矩陣中的有效位置。因此，如果輸入類似於 -

並且 k 為 1，則輸出將是 -

為了解決此問題，我們將按照以下步驟進行操作 -

• n := 行數，m = 列數
• 定義一個矩陣 ans，其順序為 n x m
• 對於 i 在 0 到 n - 1 範圍內
  • 對於 j 在 0 到 m - 1 範圍內
    • 對於 r 在 i - k 到 i + k 範圍內
      • 對於 c 在 j - k 到 j + k 範圍內
        • 如果 r 和 c 在矩陣索引中，則
          • ans[i, j] := ans[i, j] + mat[r, c]
• 返回 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, ],]

Arnab Chakraborty

更新於：30-4-2020

354 個瀏覽量

開啟您的 職業

完成課程獲得認證

開始