在C ++中重塑矩阵

在不同的平台中，有一个非常有用的功能称为“重塑”，该功能用于将矩阵重塑为大小不同的新矩阵，但数据将相同。因此，如果我们有一个矩阵，并且想要的重整矩阵的行数和列数分别有两个值r和c。

因此，如果输入类似于[[5,10]，[15,20]]，row = 1且col = 4，则输出将为[[5，10，15，20]]

为了解决这个问题，我们将按照以下步骤操作：

• 定义阵列温度

• 定义一个大小为(rxc)的2D数组分辨率

• 计数：= 0

• 对于初始化i：= 0，当i <nums的大小时，更新（将i增加1），执行-

• 在temp的末尾插入nums [i，j]

• 对于初始化j：= 0，当j <nums [0]的大小时，更新（将j增加1），执行-

• 如果r * c不等于nums的大小，则-

• 返回数字

• 对于初始化i：= 0，当i <r时，更新（将i增加1），执行-

• 计数=计数+ 1

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

• 对于初始化j：= 0，当j <c时，更新（将j增加1），执行-

• 返回资源

例 

让我们看下面的实现以更好地理解-

#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, ],]