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 queue
import "algorithms/ds"
type ElementaryPQ struct {
a ds.ArrayList
size int
// Initial capacity
capacity int
}
func NewElementaryPQ(capacity int) ElementaryPQ {
return ElementaryPQ{make(ds.ArrayList, 0, capacity), 0, capacity}
}
func (q ElementaryPQ) Insert(a int) {
q.a = append(q.a, a)
q.size++
}
func (q ElementaryPQ) Max() (max int) {
_, max = q.max()
return
}
func (q ElementaryPQ) DeleteMax() int {
if q.Empty() {
return 0
}
ind, max := q.max()
for i := ind + 1; i < q.Size(); i++ {
q.a[i-1] = q.a[i]
}
q.size--
return max
}
func (q ElementaryPQ) Empty() bool {
return q.Size() == 0
}
func (q ElementaryPQ) Size() int {
return q.size
}
func (q ElementaryPQ) Clear() {
q.size = 0
q.a = make(ds.ArrayList, 0, q.capacity)
}
func (q ElementaryPQ) max() (ind, max int) {
for i, a := range q.a {
if a > max {
ind, max = i, a
}
}
return ind, max
}
|
