​LeetCode刷题实战341:扁平化嵌套列表迭代器

共 2503字,需浏览 6分钟

 ·

2021-08-05 01:34

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !

今天和大家聊的问题叫做 扁平化嵌套列表迭代器,我们先来看题面:
https://leetcode-cn.com/problems/flatten-nested-list-iterator/


示例


示例 1

输入:nestedList = [[1,1],2,[1,1]]
输出:[1,1,2,1,1]
解释:通过重复调用 next 直到 hasNext 返回 falsenext 返回的元素的顺序应该是: [1,1,2,1,1]。

示例 2

输入:nestedList = [1,[4,[6]]]
输出:[1,4,6]
解释:通过重复调用 next 直到 hasNext 返回 falsenext 返回的元素的顺序应该是: [1,4,6]。


解题


用一个栈存储列表类型的元素,初始化时把数组从后往前压入栈,因为栈是后进先出的。函数next()就是弹出栈顶元素,然后调用getInteger()函数,将返回值返回;函数hasNext()是判断如果栈非空的话,则将栈顶元素调用isInteger()函数,如果返回true,那么返回true;否则将该元素从栈顶弹出,然后将该元素调用getList()函数,将得到的列表中的元素从后往前压入栈中。如此循环,当栈为空而退出while循环时,返回false。

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;
    }
};


好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力 。

上期推文:
LeetCode1-340题汇总,希望对你有点帮助!

浏览 34
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报