题目:![](https://i-blog.csdnimg.cn/blog_migrate/aa3ad4c669c8287c4babb6d574a69421.png)
算法思想:先判断不同矩阵的元素个数是否和原矩阵相同,然后遍历构造即可。
代码 :
vector<vector<int> > matrixReshape(vector<vector<int> >& nums, int r, int c) {
if(r * c != nums.size() * nums[0].size())
return nums;
vector<vector<int > > result(r);
vector<vector<int > > ::iterator iter;
vector<int> tmp;
int row = 0;
int column = 0;
for(int i = 0;i < nums.size();i++)
{
for(int j = 0;j < nums[0].size();j++)
{
if(column < c)
{
result[row].push_back(nums[i][j]);
column++;
}
else
{
row++;
column = 1;
result[row].push_back(nums[i][j]);
}
}
}
return result;
}