扁平化嵌套列表迭代器。给你一个嵌套的整数列表 nestedList 。每个元素要么是一个整数,要么是一个列表;该列表的元素也可能是整数或者是其他列表。请你实现一个迭代器将其扁平化,使之能够遍历这个列表中的所有整数。实现扁平迭代器类 NestedIterator :NestedIterator(List nestedList) 用嵌套列表 nestedList 初始化迭代器。int next() 返回嵌套列表的下一个整数。boolean hasNext() 如果仍然存在待迭代的整数,返回 true ;否则,返回 false 。力扣341。
自然智慧即可。最容易想到的是递归和栈。
代码用golang编写。代码如下:
type NestedIterator struct {
// 将列表视作一个队列,栈中直接存储该队列
stack [][]*NestedInteger
}
func Constructor(nestedList []*NestedInteger) *NestedIterator {
return &NestedIterator{[][]*NestedInteger{nestedList}}
}
func (it *NestedIterator) Next() int {
// 由于保证调用 Next 之前会调用 HasNext,直接返回栈顶列表的队首元素,将其弹出队首并返回
queue := it.stack[len(it.stack)-1]
val := queue[0].GetInteger()
it.stack[len(it.stack)-1] = queue[1:]
return val
}
func (it *NestedIterator) HasNext() bool {
for len(it.stack) > 0 {
queue := it.stack[len(it.stack)-1]
if len(queue) == 0 { // 当前队列为空,出栈
it.stack = it.stack[:len(it.stack)-1]
continue
}
nest := queue[0]
if nest.IsInteger() {
return true
}
// 若队首元素为列表,则将其弹出队列并入栈
it.stack[len(it.stack)-1] = queue[1:]
it.stack = append(it.stack, nest.GetList())
}
return false
}