这个题目不难,但是这里涉及了二维数组的基础知识,值得记录。
什么时候对二维数组必须初始化:在使用push_back的时候Push的不是一个数组(而是一个元素)的时候。
因为这个时候你知道怎么分区!!所有的元素都是一股脑堆在一起。
*************************************************************************************************************************************************
class Solution {
public:
vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
vector<int> t;
//对二维数组一定要进行初始化
//初始化一个r行,每行的数据是这样的:vector<int>(c,0) 二维数组。
// vector<vector<int>> newA(r,vector<int>(c,0));
//对二维数组进行初始化
vector<vector<int>> newA;
newA.resize(r);
for(int i=0;i<r;i++) newA[i].resize(c);
int count=nums[0].size();
if(count*nums.size()!=r*c)
{
return nums;
}
if(count*nums.size()==r*c)
{
//把原矩阵中的所有的元素存入到临时容器t中
for(int i=0;i<nums.size();i++)
{
for(int j=0;j<nums[0].size();j++)
{
t.push_back(nums[i][j]);
}
}
//开始填充到新矩阵
for(int i=0;i<r;i++)
{
for(int j=0;j<c;j++)
{
newA[i][j]=t[i*c+j];
}
}
return newA;
}
}
};