#ЛитКод
Задача: 341. Flatten Nested List IteratorВам дан вложенный список целых чисел 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
}
}👉
Новости 👉
База вопросов
