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

2021年3月23日

没有代码提示的我快死了

就是个遍历……只要知道函数和方法就没什么难度

注意尝试迭代，不要用递归

class NestedIterator {
private:
    int curInteger;
    stack<pair<vector<NestedInteger>::iterator, vector<NestedInteger>::iterator>> st;

public:
    NestedIterator(vector<NestedInteger> &nestedList) {
        st.emplace(nestedList.begin(),nestedList.end());
    }
    
    int next() {
        return curInteger;
    }
    
    bool hasNext() {
        while(!st.empty()){
            auto &i=st.top();
            if(i.first==i.second){
                st.pop();
                continue;
            }
            if(i.first->isInteger()){
                curInteger=i.first->getInteger();
                i.first++;
                return true;
            }
            auto &lit=i.first->getList();
            st.emplace(lit.begin(),lit.end());
            i.first++;
        }
        return false;
    }
};