Re: [閒聊] 每日leetcode

作者: nh60211as   2024-09-01 11:19:15
2022. Convert 1D Array Into 2D Array
這次算簡單,
但是要判斷無法建立 2D array 的要求滿多餘的。
練習用 iterator 寫寫看
class Solution {
public:
vector<vector<int>> construct2DArray(vector<int>& original, int m, int n)
{
if (original.size() != m * n) {
return vector<vector<int>>();
}
vector<vector<int>> result;
for (int i = 0; i < m; i++) {
int startIndex = i * n;
int endInex = (i + 1) * n; // exclusive index
vector<int> subArray(next(original.cbegin(), startIndex),
next(original.cbegin(), endInex));
result.emplace_back(subArray);
}
return result;
}
};

Links booklink

Contact Us: admin [ a t ] ucptt.com