341. 扁平化嵌套列表迭代器

给你一个嵌套的整型列表。请你设计一个迭代器，使其能够遍历这个整型列表中的所有整数。

列表中的每一项或者为一个整数，或者是另一个列表。其中列表的元素也可能是整数或是其他列表。

示例 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]。

解答

使用栈模拟深度优先搜索（取代递归），每遍历到一个嵌套列表就入栈（isInteger()返回false），直到遍历到一个数字：

class NestedIterator {
    stack<pair<vector<NestedInteger>::iterator, vector<NestedInteger>::iterator>> s;
public:
    NestedIterator(vector<NestedInteger> &nestedList) {
        s.emplace(nestedList.begin(), nestedList.end());
    }
    
    int next() {
    	// 注意迭代器要累加
        return s.top().first++->getInteger();
    }
    
    bool hasNext() {
        while(!s.empty()){
            auto &p = s.top();
            // 遍历完一个嵌套列表
            if(p.first == p.second){
                s.pop();
                continue;
            }
            // 遍历到数字
            if(p.first->isInteger()){
                return true;
            }
            // 遍历到新的嵌套列表
            else{
                auto &nested = p.first->getList();
                // 入栈新的嵌套列表
                // 记得将此前的迭代器累加指向这个新的嵌套列表的后一个元素
                p.first++;
                s.emplace(nested.begin(), nested.end());
            }
        }
        return false;
    }
};