class Solution {
public:
vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
int x=0;
int y=0;
vector<vector<int>> n(r);
if(mat.size()*mat[0].size()!=r*c)
{
return mat;
}
for(int i=0;i<r;i++)
{
n[i].resize(c);
for(int j=0;j<c;j++)
{
n[i][j]=mat[x][y];
++y;
if(y==mat[0].size())
{
++x;
y=0;
}
}
}
return n;
}
};
花了二十分钟,争取将来简单题都能在十分钟内解决!
加油!