[Swift]LeetCode341. 压平嵌套链表迭代器 $ Flatten Nested List Iterator

Given a nested list of integers, implement an iterator to flatten it.

Each element is either an integer, or a list -- whose elements may also be integers or other lists.

Example 1:
Given the list [[1,1],2,[1,1]],

By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,1,2,1,1].

Example 2:
Given the list [1,[4,[6]]],

By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,4,6].


给定一个嵌套的整数列表,实现一个迭代器来展平它。

每个元素要么是一个整数,要么是一个列表——其元素也可以是整数或其他列表。

例1:

给出列表[[1,1],2,[1,1]],

通过反复调用next,直到hasNext返回false,next返回的元素顺序应该是:[1,1,2,1,1]。

例2:

给出清单[1,[4,[6]]],

通过重复调用next直到hasNext返回false,next返回的元素顺序应该是:[1,4,6]。


 Solution:

 1 class NestedIterator
 2 {
 3     var d:[NestedInteger] = [NestedInteger]()
 4     init(_ nestedList:inout [NestedInteger])
 5     {
 6         for a in nestedList
 7         {
 8             d.append(a)
 9         }
10     }
11     
12     func next() -> Int
13     {
14         var t:NestedInteger = d.removeFirst()
15         return t.getInteger()
16     }
17     
18     func hasNext() -> Bool
19     {
20         while(!d.isEmpty)
21         {
22             var t:NestedInteger = d.first!
23             if t.isInteger()
24             {
25                 return true
26             }
27             d.removeFirst()
28             for i in 0..<t.get.getList().count
29             {
30                 d.insert(t.getList()[i],at:i)
31             }
32         }
33         return false
34     }
35 }

猜你喜欢

转载自www.cnblogs.com/strengthen/p/10738395.html
今日推荐