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

题目：

给你一个嵌套的整数列表 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 }

双端队列：

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

相关推荐