文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

請設(shè)計并實現(xiàn)一個能夠展開二維向量的迭代器。該迭代器需要支持 next 和 hasNext 兩種操作。、

示例： Vector2D iterator = new Vector2D([[1,2],[3],[4]]);iterator.next(); // 返回 1 iterator.next(); // 返回 2 iterator.next(); // 返回 3 iterator.hasNext(); // 返回 true iterator.hasNext(); // 返回 true iterator.next(); // 返回 4 iterator.hasNext(); // 返回 false

注意：

• 請記得 重置 在 Vector2D 中聲明的類變量（靜態(tài)變量），因為類變量會 在多個測試用例中保持不變，影響判題準(zhǔn)確。請 查閱 這里。
• 你可以假定 next() 的調(diào)用總是合法的，即當(dāng) next() 被調(diào)用時，二維向量總是存在至少一個后續(xù)元素。

來源：力扣（LeetCode） 鏈接：https://leetcode-cn.com/problems/flatten-2d-vector
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請注明出處。

2. 解題

class Vector2D {vector<vector<int>> arr;int r = 0, c = 0, m; public:Vector2D(vector<vector<int>>& v) {arr = v;m = arr.size();}int next() {while(c == arr[r].size())//跳過沒有剩余元素的行{r++;c = 0;}return arr[r][c++];}bool hasNext() {while(r < m && c == arr[r].size()){ //跳過沒有剩余元素的行r++;c = 0;}return r < m-1 || (r == m-1 && c < arr[m-1].size());} };

72 ms 21.9 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關(guān)注我的公眾號（Michael阿明），一起加油、一起學(xué)習(xí)進步！

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎勵來咯，堅持創(chuàng)作打卡瓜分現(xiàn)金大獎

總結(jié)

以上是生活随笔為你收集整理的LeetCode 251. 展开二维向量的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。