-
Notifications
You must be signed in to change notification settings - Fork 0
/
review300.go
60 lines (53 loc) · 1.75 KB
/
review300.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package main
/**
* // This is the interface that allows for creating nested lists.
* // You should not implement it, or speculate about its implementation
* type NestedInteger struct {
* }
*
* // Return true if this NestedInteger holds a single integer, rather than a nested list.
* func (this NestedInteger) IsInteger() bool {}
*
* // Return the single integer that this NestedInteger holds, if it holds a single integer
* // The result is undefined if this NestedInteger holds a nested list
* // So before calling this method, you should have a check
* func (this NestedInteger) GetInteger() int {}
*
* // Set this NestedInteger to hold a single integer.
* func (n *NestedInteger) SetInteger(value int) {}
*
* // Set this NestedInteger to hold a nested list and adds a nested integer to it.
* func (this *NestedInteger) Add(elem NestedInteger) {}
*
* // Return the nested list that this NestedInteger holds, if it holds a nested list
* // The list length is zero if this NestedInteger holds a single integer
* // You can access NestedInteger's List element directly if you want to modify it
* func (this NestedInteger) GetList() []*NestedInteger {}
*/
type NestedIterator struct {
stack []int
}
func Constructor(nestedList []*NestedInteger) *NestedIterator {
stack := []int{}
helper(nestedList, &stack)
return &NestedIterator{
stack,
}
}
func (this *NestedIterator) Next() int {
head := this.stack[0]
this.stack = this.stack[1:]
return head
}
func (this *NestedIterator) HasNext() bool {
return len(this.stack) > 0
}
func helper(nestedList []*NestedInteger, stack *[]int) {
for i := range nestedList {
if nestedList[i].IsInteger() {
*stack = append(*stack, nestedList[i].GetInteger())
} else {
helper(nestedList[i].GetList(), stack)
}
}
}