diff --git a/2201/220101-CN.cpp b/2201/220101-CN.cpp index 71937db..8a3bf72 100644 --- a/2201/220101-CN.cpp +++ b/2201/220101-CN.cpp @@ -1,4 +1,27 @@ -// -// Created by Lam Haoyin on 01/01/22. -// +#include +#include +class Solution { +public: + static std::vector> construct2DArray(std::vector& original, int m, int n) { + if (m * n != original.size()) return {}; + std::vector> ret; + ret.reserve(m); + for (int i = 0; i < m; ++i) { + ret.emplace_back(original.begin() + (i * n), original.begin() + ((1 + i) * n)); + } + return ret; + } +}; + +int main() { + std::vector s {1, 2}; + const auto ret = Solution::construct2DArray(s, 1, 1); + for (const auto& i : ret) { + for (const auto& j : i) { + std::printf("%d ", j); + } + std::printf("\n"); + } + return 0; +}