文章目錄

• • 1. 題目
  • 2. 雙棧解題

1. 題目

給定一個嵌套的整型列表。設計一個迭代器，使其能夠遍歷這個整型列表中的所有整數。

列表中的項或者為一個整數，或者是另一個列表。

示例 1: 輸入: [[1,1],2,[1,1]] 輸出: [1,1,2,1,1] 解釋: 通過重復調用 next 直到 hasNext 返回false， next 返回的元素的順序應該是: [1,1,2,1,1]。示例 2: 輸入: [1,[4,[6]]] 輸出: [1,4,6] 解釋: 通過重復調用 next 直到 hasNext 返回false， next 返回的元素的順序應該是: [1,4,6]。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/flatten-nested-list-iterator
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 雙棧解題

• 建立兩個棧，一個存儲起點迭代器，一個存儲終點迭代器
• 如果兩個棧頂相等，說明當前list遍歷完了，兩棧都彈棧
• 如果棧不為空，且棧頂不等，起點棧頂是數字嗎，是數字，可以打印了，然后移動迭代器
• 是列表，需要先把起點棧頂移動一位，然后再將移動前的迭代器（指向列表）對應的起點終點分別壓棧，后面優先處理該列表

/*** class NestedInteger {* public:* bool isInteger() const;* int getInteger() const;* const vector<NestedInteger> &getList() const;* };*/class NestedIterator {stack<vector<NestedInteger>::iterator> begins, ends; public:NestedIterator(vector<NestedInteger> &nestedList) {begins.push(nestedList.begin());ends.push(nestedList.end());}int next() {hasNext();return (begins.top()++)->getInteger();}bool hasNext() {vector<NestedInteger>::iterator tp;while(!begins.empty()){if(begins.top()==ends.top()){begins.pop();ends.pop();}else{tp = begins.top();if(tp->isInteger())return true;//不是整數,是個列表begins.top()++;//當前的移到下一個//處理tp這個列表，壓棧begins.push(tp->getList().begin());ends.push(tp->getList().end());;}}return false;} };

總結

以上是生活随笔為你收集整理的LeetCode 341. 扁平化嵌套列表迭代器（双栈）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。