要素の合計がC++の特定の値に等しい行列から部分行列の数を見つけるプログラム

整数値を含む行列が与えられたと仮定します。行列の要素の合計が特定のターゲット合計値に等しい行列から部分行列を見つける必要があります。部分行列の数を返します。

したがって、入力が次のような場合

ターゲット=5の場合、出力は3になります。

要素の合計が6に等しい部分行列の数は2です。

これを解決するには、次の手順に従います-

• n：=マットのサイズ
• m：=（nが0と同じ場合は0、それ以外の場合はmat [0]のサイズ）
• m> nの場合、-
  • 次元mxnの1つの2D配列転置を定義します
  • iを初期化する場合：=0、i
  • jを初期化する場合：=0、j
  • transpose [j、i]：=mat [i、j]

例

理解を深めるために、次の実装を見てみましょう-

#include<bits/stdc++.h>

using namespace std;

int solve(vector<vector<int>>& mat, int sumTarget) {
   int n = mat.size();
   int m = n == 0 ? 0 : mat[0].size();

   if (m > n) {
      vector<vector<int>> transpose(m, vector<int>(n));
      for (int i = 0; i < n; i++) {
         for (int j = 0; j < m; j++) {
            transpose[j][i] = mat[i][j];
         }
      }
      return solve(transpose, sumTarget);
   }

   int ans = 0;
   for (int p = 0; p < m; p++) {
      vector<int> arr(n);
      for (int q = p; q < m; q++) {
         for (int i = 0; i < n; i++) {
            arr[i] += mat[i][q];
         }

         unordered_map<int, int> pcnt = {{0, 1}};
         int pref = 0;
         for (int i = 0; i < n; i++) {
            pref += arr[i];
            auto tmp = pcnt.find(pref - sumTarget);
            if (tmp != end(pcnt)) ans += tmp->second;
            pcnt[pref]++;
         }
      }
   }
   return ans;
}

int main() {
   vector<vector<int>> mat = {{0, 0, 1, 0}, {0, 1, 0, 0}, {0, 1, 0, 1}, {1, 1, 0, 1}};
cout<< solve(mat, 5) <<endl;
return 0;
}

入力

{{0, 0, 1, 0}, {0, 1, 0, 0}, {0, 1, 0, 1}, {1, 1, 0, 1}}, 5

出力

3