给你一个嵌套的整型列表。请你设计一个迭代器,使其能够遍历这个整型列表中的所有整数。
列表中的每一项或者为一个整数,或者是另一个列表。其中列表的元素也可能是整数或是其他列表。
示例 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]。
提示: 接口如下:
* // This is the interface that allows for creating nested lists. * // You should not implement it, or speculate about its implementation * class NestedInteger { * public: * // Return true if this NestedInteger holds a single integer, rather than a nested list. * bool isInteger() const; * * // Return the single integer that this NestedInteger holds, if it holds a single integer * // The result is undefined if this NestedInteger holds a nested list * int getInteger() const; * * // Return the nested list that this NestedInteger holds, if it holds a nested list * // The result is undefined if this NestedInteger holds a single integer * const vector<NestedInteger> &getList() const; * };思路:
使用栈,记录每个列表。对于每个列表,我们判断它是否是单个整数,使用接口中的函数bool isInteger() const 。若它包含子列表,则将子列表也压入栈中。代码: 方法1:
class NestedIterator { public: stack<NestedInteger*> st; NestedIterator(vector<NestedInteger> &nestedList) { for(int i=nestedList.size()-1;i>=0;--i) st.push(&nestedList[i]); } int next() { //返回单个整数 int res = st.top()->getInteger(); st.pop(); return res; } bool hasNext() { //记录子列表 if(st.empty()) return false; NestedInteger* tmp = st.top(); while(!tmp->isInteger()){ st.pop(); vector<NestedInteger> &list = tmp->getList(); for(int i = list.size()-1;i>=0;--i) st.push(&list[i]); if(st.empty()) return false; tmp = st.top(); } return true; } };方法2:
class NestedIterator { public: vector<int> data; vector<int>::iterator it; NestedIterator(vector<NestedInteger> &nestedList) { parse(nestedList); it = data.begin(); } void parse(vector<NestedInteger> &nestedList){ for(auto nes : nestedList){ if(nes.isInteger()) data.push_back(nes.getInteger()); else parse(nes.getList()); } } int next() { return *it++; } bool hasNext() { return it != data.end(); } };结果:
参考链接: [1] Zolewit:341. 扁平化嵌套列表迭代器/C++