原题地址: https://leetcode.com/problems/flatten-nested-list-iterator/
给定一个嵌套的整型列表。设计一个迭代器,使其能够遍历这个整型列表中的所有整数。
列表中的项或者为一个整数,或者是另一个列表。
示例 1:
输入: [[1,1],2,[1,1]]
输出: [1,1,2,1,1]
解释: 通过重复调用 next 直到 hasNext 返回false,next 返回的元素的顺序应该是: [1,1,2,1,1]。
示例 2:
输入: [1,[4,[6]]]
输出: [1,4,6]
解释: 通过重复调用 next 直到 hasNext 返回false,next 返回的元素的顺序应该是: [1,4,6]。
这个其实不难,其实就是一个嵌套的树展开的问题。我甚至都没用到堆栈,按理说这是一道堆栈相关的题目,但是我没明白有啥必须用堆栈的地方,就是写个递归函数flatten来展开树而已。
遇到数字节点就直接扔进列表,遇到嵌套节点,就调用flatten嵌套解决。
Github 地址: https://github.com/tinyfool/leetcode/tree/master/src/p0341
其他堆栈和队列相关题目,参照堆栈和队列专题。