如果要返回的matrix元素个数与原先不同,直接返回。 否则遍历一遍原matrix, 重新组装成一个新的行列数的matrix
vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c)
{
int pr = nums.size();
int pc = nums[0].size();
if (r*c != pr*pc)
return nums;
vector<vector<int>> vec_ret;
vector<int> vec_temp;
int count = 0;
for (int row = 0; row < pr; ++row)
for (int col = 0; col < pc; ++col)
{
vec_temp.push_back(nums[row][col]);
if ((count + 1) % c == 0)
{
vec_ret.push_back(vec_temp);
vec_temp = vector<int>();
}
++count;
}
return vec_ret;
}