【力扣·每日一题】2022. 将一维数组转变成二维数组（C++ 模拟 vector）

linkk

题意

思路

只用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;
    }
};