資料結構
網路
RDBMS
作業系統
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP
物理學
化學
生物學
數學
英語
經濟
心理學
社會學
時尚學
法律研究用 C++ 改變矩陣形狀
在不同的平臺上，有一個非常有用的函式稱為“重新整形”，該函式用於將矩陣重新整形為具有不同大小的新矩陣，但資料將保持相同。因此，如果我們有一個矩陣和兩個值 r 和 c，分別表示所需重新整形矩陣的行數和列數。
因此，如果輸入類似 [[5,10],[15,20]]，row = 1 且 col = 4，則輸出將為 [[5, 10, 15, 20]]
為了解決這個問題，我們將遵循以下步驟-
定義一個數組 temp
定義一個大小為 (r x c) 的二維陣列 res
count := 0
對 i 初始化 := 0，當 i < size of nums，更新（增加 i 1），執行 -
對 j 初始化 := 0，當 j < size of nums[0]，更新（增加 j 1），執行 -
在 temp 的末尾插入 nums[i, j]
如果 r * c 不等於 nums 的大小，則 -
返回 nums
對 i 初始化 := 0，當 i < r，更新（增加 i 1），執行 -
對 j 初始化 := 0，當 j < c，更新（增加 j 1），執行 -
count = count + 1
res[i, j] := temp[count]
返回 res
示例
讓我們看看下面的實現，以獲得更好的理解 -
#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, ],]
更新於： 11-6-2020
1K+ 檢視
廣告