341. Flatten Nested List Iterator

最新推荐文章于 2022-05-08 19:51:35 发布

tbabm_

最新推荐文章于 2022-05-08 19:51:35 发布

阅读量102

点赞数

分类专栏：算法文章标签： stl

本文链接：https://blog.csdn.net/hdjiguangxi/article/details/89474659

版权

算法专栏收录该内容

47 篇文章 0 订阅

订阅专栏

设计一个迭代器，处理包含整数或嵌套列表的结构，按顺序返回所有元素。题目给出了两个示例，需要遍历列表并递归处理子列表。

摘要由CSDN通过智能技术生成

Given a nested list of integers, implement an iterator to flatten it.

Each element is either an integer, or a list -- whose elements may also be integers or other lists.

Example 1:

Input: [[1,1],2,[1,1]]
Output: [1,1,2,1,1]
Explanation: By calling next repeatedly until hasNext returns false, 
             the order of elements returned by next should be:[1,1,2,1,1]

Example 2:

Input: [1,[4,[6]]]
Output: [1,4,6]
Explanation: By calling next repeatedly until hasNext returns false, 
             the order of elements returned by next should be: [1,4,6]

题意：给出一个嵌套的节点列表，其中每个节点要么为一个整数，要么为一个列表。要求为每次输入的列表实现一个迭代器，能够按序输出其中的元素。

思路：从第一个列表开始遍历其每个元素，若为整型，存放到队列中，否则递归求解该子列表。

class NestedIterator {
public:
    queue<int>que;
    void solve(vector<NestedInteger> & tmp){
        for(int i=0;i<tmp.size();i++)
        {
          if(tmp[i].isInteger())
                que.push(tmp[i].getInteger());
          else{
                solve(tmp[i].getList());
          }
        }
    }
    
    NestedIterator(vector<NestedInteger> &nestedList) {
         for(int i=0;i<nestedList.size();i++){
              if(nestedList[i].isInteger()){
                   que.push(nestedList[i].getInteger());
              }else{
                  solve(nestedList[i].getList());
              } 
         }
    }

    int next() {
        int tmp=que.front();
        que.pop();
        return tmp;
    }

    bool hasNext() {
        return que.size()==0?0:1;
    }
};