1、先判断是否能构成想要的二维矩阵,不能则返回一个空二维数组,能则用两重循环给二维数组一一赋值。
class Solution {
public:
vector<vector<int>> construct2DArray(vector<int>& original, int m, int n) {
int thesize=original.size(),count=0;
if(m*n!=thesize)
{
vector<vector<int> > vec;
return vec;
}
else
{
vector<vector<int> > vec(m,vector<int>(n));
for(int i=0;i<m;++i)
for(int j=0;j<n;++j)
vec[i][j]=original[count++];
return vec;
}
}
};