力扣341(java)-扁平化嵌套列表迭代器（中等）

2022-09-11 15:04:37 阅读：240 来源： 互联网

标签：遍历 hasNext 整数列表力扣 next 341 java nestedList

题目：

给你一个嵌套的整数列表 nestedList 。每个元素要么是一个整数，要么是一个列表；该列表的元素也可能是整数或者是其他列表。请你实现一个迭代器将其扁平化，使之能够遍历这个列表中的所有整数。

实现扁平迭代器类 NestedIterator ：

NestedIterator(List<NestedInteger> nestedList) 用嵌套列表 nestedList 初始化迭代器。
int next() 返回嵌套列表的下一个整数。
boolean hasNext() 如果仍然存在待迭代的整数，返回 true ；否则，返回 false 。

你的代码将会用下述伪代码检测：

initialize iterator with nestedList
res = []
while iterator.hasNext()
append iterator.next() to the end of res
return res

如果 res 与预期的扁平化列表匹配，那么你的代码将会被判为正确。

示例 1：

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

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

提示：

1 <= nestedList.length <= 500
嵌套列表中的整数值在范围 [-106, 106] 内

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/flatten-nested-list-iterator
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：

解释中给出了三种方法：

isInteger() // 判断元素是否为数字
getInteger() // 获取数字元素
getList() // 获取非数字元素

调用方法：

/**
* Your NestedIterator will be called like this:
* var i = new NestedIterator(nestedList), a = [];
* while (i.hasNext()) a.push(i.next());
*/

一、栈

将列表中的元素从后往前的顺序放入栈中，这样列表中头元素就在栈顶。

每次调用next()时需要判断 hasNext()返回的结果，判断下一个对象是否为整数：

如果是整数：将栈顶元素弹出，并返回该整数；
如果是列表：将该列表中的元素从后往前遍历，依次放入栈中，然后递归判断栈顶元素是否为整数。

注意：

1.栈中存放的类型为：NestedInteger，因为不仅可能为整数还可能为一个整数

2.--i：先往前移一位再使用，例如[1,2,[1,1]],先遍历到[1,1],然后--i，遍历到2...

代码：

 1 public class NestedIterator implements Iterator<Integer> {
 2     Deque<NestedInteger> stack = new ArrayDeque<>();
 3 
 4     public NestedIterator(List<NestedInteger> nestedList) {
 5         //从后往前遍历，存放入栈，保证表头在栈顶
 6         for(int i = nestedList.size()-1; i >= 0; --i){
 7             stack.addLast(nestedList.get(i));
 8         } 
 9     }
10 
11     @Override
12     public Integer next() {
13         return hasNext() ? stack.pollLast().getInteger() : -1; //如果还存在迭代的整数，就取栈顶中的数字
14         
15     }
16 
17     @Override
18     public boolean hasNext() {
19         if(stack.isEmpty()){
20             return false;
21         }else{
22             //如果栈顶为整数
23             NestedInteger temp = stack.peekLast();
24             if(temp.isInteger()){
25                 return true;
26             }else{
27                 //说明栈顶是列表，将列表倒序放入栈，然后递归检查
28                 List<NestedInteger> list = stack.pollLast().getList();
29                 for(int i = list.size() - 1; i >= 0; --i){
30                     stack.addLast(list.get(i));
31                 }
32                 return hasNext();
33             }
34         }
35         
36     }
37 }

二、DFS(深度优先搜索)

eg: [[2,3], 5,1,[4,2,1]]

先遍历[2,3]，发现其是嵌套列表，再遍历2，3，然后依次遍历5，1，再遍历[4,2,1],发现其是嵌套列表，再遍历4，2，1。

利用队列的先进先出的特点，通过递归的方式，将所有的元素压入队列中，next()时，如果队列中有元素直接从队首弹出，否则返回-1。

代码：

 1 public class NestedIterator implements Iterator<Integer> {
 2     Deque<Integer> queue = new ArrayDeque<>();
 3 
 4     public NestedIterator(List<NestedInteger> nestedList) {
 5         dfs(nestedList);
 6     }
 7 
 8     @Override
 9     public Integer next() {
10         return hasNext() ? queue.pollFirst() : -1;
11         
12     }
13 
14     @Override
15     public boolean hasNext() {
16        return !queue.isEmpty();    
17     }
18     void dfs(List<NestedInteger> list){
19         for(NestedInteger temp : list){
20             if(temp.isInteger()){
21                 queue.addLast(temp.getInteger());
22             }else{
23                 dfs(temp.getList());
24             }
25         }
26     }
27 }

双端队列：

标签：遍历,hasNext,整数,列表,力扣,next,341,java,nestedList
来源： https://www.cnblogs.com/liu-myu/p/16683749.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

力扣341(java)-扁平化嵌套列表迭代器（中等）