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