?LeetCode刷題實戰(zhàn)341：扁平化嵌套列表迭代器

示例 1：

輸入：nestedList = [[1,1],2,[1,1]]
輸出：[1,1,2,1,1]
解釋：通過重復調(diào)用 next 直到 hasNext 返回 false，next 返回的元素的順序應該是: [1,1,2,1,1]。

示例 2：

輸入：nestedList = [1,[4,[6]]]
輸出：[1,4,6]
解釋：通過重復調(diào)用 next 直到 hasNext 返回 false，next 返回的元素的順序應該是: [1,4,6]。

class NestedIterator {
private:
    stack<NestedInteger> s;
    
public:
    NestedIterator(vector<NestedInteger> &nestedList) {
        for(int i = nestedList.size() - 1; i >= 0; i--)
            s.push(nestedList[i]);
    }
 
    int next() {
        NestedInteger temp = s.top();
        s.pop();
        return temp.getInteger();
    }
 
    bool hasNext() {
        while(!s.empty()){
            NestedInteger temp = s.top();
            if(temp.isInteger())
                return true;
            s.pop();
            for(int i = temp.getList().size() - 1; i >= 0; i--)
                s.push(temp.getList()[i]);
        }
        return false;
    }
};