题意
思路
只用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; } };