题意
思路
只用vector的基本操作push_back和size就可以完成本题。
用二维vector存储答案,tmp表示一行的数。
遍历给出的original数组,将元素放到tmp里
每次对tmp里的元素个数进行判断,如果tmp的元素个数等于n的话,说明应该放到下一行了,就将当前tmp放入ans里,清空tmp。
代码
class Solution {
public:
vector<vector<int>> construct2DArray(vector<int>& original, int m, int n) {
vector<vector<int>>ans;
if(n*m!=original.size()) return ans;
vector<int>tmp;
for(int i=0;i<original.size();i++){
tmp.push_back(original[i]);
if(tmp.size()==n) ans.push_back(tmp),tmp.clear();
}
return ans;
}
};