Функция heap.Push

Функция heap.Push добавляет новый элемент в кучу, сохраняя ее свойства. Она принимает два параметра: первый — это интерфейс heap.Interface, который представляет кучу, второй — элемент, который нужно добавить. Функция автоматически корректирует структуру кучи, чтобы она оставалась упорядоченной.

Синтаксис

goheap.Push(h, x)

Пример

Давайте создадим кучу из целых чисел и добавим в нее новый элемент:

gopackage main

import (
  "container/heap"
  "fmt"
)

type IntHeap []int

func (h IntHeap) Len() int       { return len(h) }
func (h IntHeap) Less(i, j int) bool { return h[i] < h[j] }
func (h IntHeap) Swap(i, j int)    { h[i], h[j] = h[j], h[i] }

func (h *IntHeap) Push(x any) {
  *h = append(*h, x.(int))
}

func (h *IntHeap) Pop() any {
  old := *h
  n := len(old)
  x := old[n-1]
  *h = old[0 : n-1]
  return x
}

func main() {
  h := &IntHeap{1, 3, 5, 7, 9}
  heap.Init(h)
  heap.Push(h, 4)
  fmt.Println(*h)
}

Результат выполнения кода:

go[1 3 4 7 9 5]

Пример

Давайте создадим кучу из строк и добавим в нее новый элемент:

gopackage main

import (
  "container/heap"
  "fmt"
)

type StringHeap []string

func (h StringHeap) Len() int       { return len(h) }
func (h StringHeap) Less(i, j int) bool { return h[i] < h[j] }
func (h StringHeap) Swap(i, j int)    { h[i], h[j] = h[j], h[i] }

func (h *StringHeap) Push(x any) {
  *h = append(*h, x.(string))
}

func (h *StringHeap) Pop() any {
  old := *h
  n := len(old)
  x := old[n-1]
  *h = old[0 : n-1]
  return x
}

func main() {
  h := &StringHeap{"abcde", "bcd", "cdef"}
  heap.Init(h)
  heap.Push(h, "abc")
  fmt.Println(*h)
}

Результат выполнения кода:

go[abc abcde bcd cdef]

Смотрите также