566. Reshape the Matrix

题目

解题思路

具体代码

class Solution {
public:
vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {

int origin_r = nums.size(), origin_c = 0;
if (origin_r)
origin_c = nums[0].size();
if (origin_r * origin_c != r * c)
return nums;

vector<int> tmp;
for (int i = 0; i < origin_r; i++)
for (int j = 0; j < origin_c; j++) {
tmp.push_back(nums[i][j]);

}

vector<vector<int>> nums_t;
for (int i = 0; i < r; i++) {
vector<int> tmp_t;
for (int j = 0; j < c; j++) {
tmp_t.push_back(tmp[i * c + j]);
}
nums_t.push_back(tmp_t);
}

return nums_t;
}
};