summaryrefslogtreecommitdiff
path: root/queue/elementarypriority.go
blob: 508cbcb1a5e40b1665b80cd79bcd5fcbb1258cae (plain)
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 ElementaryPriority struct {
	a    ds.ArrayList
	size int
	// Initial capacity
	capacity int
}

func NewElementaryPriority(capacity int) *ElementaryPriority {
	return &ElementaryPriority{make(ds.ArrayList, 0, capacity), 0, capacity}
}

func (q *ElementaryPriority) Insert(a int) {
	q.a = append(q.a, a)
	q.size++
}

func (q *ElementaryPriority) Max() (max int) {
	_, max = q.max()
	return
}

func (q *ElementaryPriority) 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 *ElementaryPriority) Empty() bool {
	return q.Size() == 0
}

func (q *ElementaryPriority) Size() int {
	return q.size
}

func (q *ElementaryPriority) Clear() {
	q.size = 0
	q.a = make(ds.ArrayList, 0, q.capacity)
}

func (q *ElementaryPriority) max() (ind, max int) {
	for i, a := range q.a {
		if a > max {
			ind, max = i, a
		}
	}
	return ind, max
}