Convert 1D Array Into 2D Array LeetCode solution in C++

1
class Solution {
2
 public:
3
  vector<vector<int>> construct2DArray(vector<int> &original, int m, int n) {
4
    if (m * n != original.size()) return {};
5

6
    vector<vector<int>> res;
7
    for (int i = 0; i < m * n; i += n)
8
      res.push_back(vector<int>(original.begin() + i, original.begin() + i + n));
9

10
    return res;
11
  }
12
};