-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathqueue.go
90 lines (79 loc) · 1.31 KB
/
queue.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
package main
type Queue[D Data] struct {
head *QNode[D]
tail *QNode[D]
length int
}
type QNode[D Data] struct {
val D
next *QNode[D]
}
func newQNode[D Data](val D, next *QNode[D]) *QNode[D] {
return &QNode[D]{val: val, next: next}
}
func NewQueue[D Data]() *Queue[D] {
return &Queue[D]{}
}
func (q *Queue[D]) Add(data D) {
node := newQNode[D](data, nil)
if q.head == nil || q.tail == nil {
q.head = node
q.tail = node
}
if q.tail != nil {
tail := q.tail
tail.next = node
q.tail = node
}
q.length++
}
func (q Queue[D]) IsEmpty() bool {
if q.length == 0 {
return true
}
return false
}
func (q Queue[D]) GetLength() int {
return q.length
}
func (q Queue[D]) Contains(data D) (bool, int) {
node := q.head
count := 0
for {
if node.val == data {
return true, count
}
if node.next == nil {
return false, -1
}
count++
node = node.next
}
}
func (q *Queue[D]) Pop() *QNode[D] {
head := q.head
q.head = head.next
head.next = nil
q.length--
return head
}
func (q Queue[D]) PrintData() {
node := q.head
count := 0
if node == nil {
println("Empty List !")
return
}
for {
if node.next == nil {
println(node.val)
return
}
print(node.val, " -> ")
count++
node = node.next
}
}
func (q Queue[D]) Peek() D {
return q.head.val
}