-
Notifications
You must be signed in to change notification settings - Fork 29
/
list.go
93 lines (79 loc) · 1.43 KB
/
list.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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
package linkedlist
type List[T any] struct {
head *Item[T]
end *Item[T]
count int
}
func NewList[T any]() *List[T] {
return &List[T]{}
}
func (this *List[T]) Head() *Item[T] {
return this.head
}
func (this *List[T]) End() *Item[T] {
return this.end
}
func (this *List[T]) Push(item *Item[T]) {
if item == nil {
return
}
// 如果已经在末尾了,则do nothing
if this.end == item {
return
}
if item.prev != nil || item.next != nil || this.head == item {
this.Remove(item)
}
this.add(item)
}
func (this *List[T]) Remove(item *Item[T]) {
if item == nil {
return
}
if item.prev != nil {
item.prev.next = item.next
}
if item.next != nil {
item.next.prev = item.prev
}
if item == this.head {
this.head = item.next
}
if item == this.end {
this.end = item.prev
}
item.prev = nil
item.next = nil
this.count--
}
func (this *List[T]) Len() int {
return this.count
}
func (this *List[T]) Range(f func(item *Item[T]) (goNext bool)) {
for e := this.head; e != nil; e = e.next {
goNext := f(e)
if !goNext {
break
}
}
}
func (this *List[T]) Reset() {
this.head = nil
this.end = nil
}
func (this *List[T]) add(item *Item[T]) {
if item == nil {
return
}
if this.end != nil {
this.end.next = item
item.prev = this.end
item.next = nil
}
this.end = item
if this.head == nil {
this.head = item
}
this.count++
}