C++でマトリックスの形を変える

別のプラットフォームには、「reshape」と呼ばれる非常に便利な関数があります。この関数は、行列を異なるサイズの新しい行列に再形成するために使用されますが、データは同じになります。したがって、行列があり、必要な再形成された行列の行番号と列番号にそれぞれ2つの値rとcがある場合。

したがって、入力が[[5,10]、[15,20]]、row =1、col =4の場合、出力は[[5、10、15、20]]

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

• アレイの温度を定義する

• サイズ（r x c）の2D配列解像度を1つ定義します

• カウント：=0

• 初期化i：=0の場合、i

  • 初期化j：=0の場合、j

    • tempの最後にnums[i、j]を挿入します

• r * cがnumsのサイズと等しくない場合、-

  • 数値を返す

• 初期化i：=0の場合、i

  • 初期化j：=0の場合、j

    • カウント=カウント+1

    • res [i、j]：=temp [count]

• 解像度を返す

例

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

#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>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
      vector<int> temp;
      vector<vector<int> > res(r, vector<int>(c));
      int count = 0;
      for (int i = 0; i < nums.size(); i++) {
         for (int j = 0; j < nums[0].size(); j++) {
            temp.push_back(nums[i][j]);
         }
      }
      if (r * c != nums.size() * nums[0].size())
         return nums;
      for (int i = 0; i < r; i++) {
         for (int j = 0; j < c; j++) {
            res[i][j] = temp[count++];
         }
      }
      return res;
   }
};
main(){
   Solution ob;
   vector<vector<int>> v = {{5,10},{15,20}};
   print_vector(ob.matrixReshape(v, 1, 4));
}

入力

{{5,10},{15,20}}, 1, 4

出力

[[5, 10, 15, 20, ],]