-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpriorityQueue.go
62 lines (43 loc) · 909 Bytes
/
priorityQueue.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
package main
import "fmt"
func heapify(slice []int, n, i int) []int {
if n == 1 {
fmt.Println("Single element")
} else {
largest := i
left := 2*i + 1
right := 2*i + 2
if left < n && slice[left] > slice[largest] {
largest = left
}
if right < n && slice[right] > slice[largest] {
largest = right
}
if largest != i {
slice[i], slice[largest] = slice[largest], slice[i]
heapify(slice, n, largest)
}
}
return slice
}
func insert(slice []int, newItem int) []int {
size := len(slice)
if size == 0 {
slice = append(slice, newItem)
} else {
slice = append(slice, newItem)
for i := size/2 - 1; i > -1; i-- {
slice = heapify(slice, size, i)
}
}
return slice
}
func main() {
var slice = []int{}
slice = insert(slice, 3)
slice = insert(slice, 5)
slice = insert(slice, 1)
slice = insert(slice, 10)
slice = insert(slice, 13)
fmt.Println(slice)
}