Сложность: medium
Вам дан вложенный список целых чисел nestedList. Каждый элемент либо является целым числом, либо списком, элементы которого также могут быть целыми числами или другими списками. Реализуйте итератор для его развёртки.
Реализуйте класс NestedIterator:
NestedIterator(List<NestedInteger> nestedList) Инициализирует итератор вложенным списком nestedList.
int next() Возвращает следующий целый элемент вложенного списка.
boolean hasNext() Возвращает true, если в вложенном списке еще остались целые числа, и false в противном случае.
Пример:
Input: nestedList = [[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].
👨💻 Алгоритм:
1⃣Инициализация
Сохраняйте исходный вложенный список в стеке или очереди. Используйте стек для сохранения состояния итерации по вложенным спискам.
2⃣Метод next()
Возвращает следующий целый элемент из стека или очереди. Если текущий элемент является списком, развёртывайте его и добавляйте элементы в стек.
3⃣Метод hasNext()
Проверяет, есть ли в стеке или очереди оставшиеся целые элементы. Если на вершине стека находится список, развёртывайте его до тех пор, пока не встретится целый элемент.
😎 Решение:
class NestedIterator {
private var stack: [NestedInteger] = []
init(_ nestedList: [NestedInteger]) {
self.flatten(nestedList)
}
private func flatten(_ nestedList: [NestedInteger]) {
for ni in nestedList.reversed() {
stack.append(ni)
}
}
func next() -> Int {
return stack.removeLast().getInteger()
}
func hasNext() -> Bool {
while !stack.isEmpty, !stack.last!.isInteger() {
flatten(stack.removeLast().getList())
}
return !stack.isEmpty
}
}Ставь 👍 и забирай 📚 Базу знаний