[英]How to extract x top int values from a map in Golang?
我有一張map[string]int
我想從中獲取x頂部值並將它們存儲在另一個數據結構,另一個映射或切片中。 來自https://blog.golang.org/go-maps-in-action#TOC_7。 我明白:
當使用范圍循環在地圖上進行迭代時,未指定迭代順序,並且不能保證每次迭代之間都相同。
因此,結果結構將是一個切片。
我看了幾個相關主題,但沒有一個適合我的問題:
請問最有效的方法是什么?
謝謝,
編輯:我的解決方案是將地圖變成一個切片並對其進行排序,然后提取第一個x值。
但是有更好的方法嗎?
package main
import (
"fmt"
"sort"
)
func main() {
// I want the x top values
x := 3
// Here is the map
m := make(map[string]int)
m["k1"] = 7
m["k2"] = 31
m["k3"] = 24
m["k4"] = 13
m["k5"] = 31
m["k6"] = 12
m["k7"] = 25
m["k8"] = -8
m["k9"] = -76
m["k10"] = 22
m["k11"] = 76
// Turning the map into this structure
type kv struct {
Key string
Value int
}
var ss []kv
for k, v := range m {
ss = append(ss, kv{k, v})
}
// Then sorting the slice by value, higher first.
sort.Slice(ss, func(i, j int) bool {
return ss[i].Value > ss[j].Value
})
// Print the x top values
for _, kv := range ss[:x] {
fmt.Printf("%s, %d\n", kv.Key, kv.Value)
}
}
如果我想在地圖的末尾有x個top值,那么根據我的解決方案,我將不得不再次將切片變成地圖。 這仍然是最有效的方法嗎?
創建切片和排序是一個很好的解決方案; 但是,您也可以使用堆 。 兩種實現方式的Big O性能應相等(n log n),因此這是一個可行的選擇,其優點是,如果要添加新條目,您仍可以有效地訪問前N個項目,而無需重復排序整個集合。
要使用堆,您可以使用一個Less
函數(將Values 大於 ( h[i].Value > h[j].Value
)進行比較)來實現kv
類型的heap.Interface
,添加映射中的所有條目,然后彈出要使用的項目數。
例如( Go Playground ):
func main() {
m := getMap()
// Create a heap from the map and print the top N values.
h := getHeap(m)
for i := 1; i <= 3; i++ {
fmt.Printf("%d) %#v\n", i, heap.Pop(h))
}
// 1) main.kv{Key:"k11", Value:76}
// 2) main.kv{Key:"k2", Value:31}
// 3) main.kv{Key:"k5", Value:31}
}
func getHeap(m map[string]int) *KVHeap {
h := &KVHeap{}
heap.Init(h)
for k, v := range m {
heap.Push(h, kv{k, v})
}
return h
}
// See https://golang.org/pkg/container/heap/
type KVHeap []kv
// Note that "Less" is greater-than here so we can pop *larger* items.
func (h KVHeap) Less(i, j int) bool { return h[i].Value > h[j].Value }
func (h KVHeap) Swap(i, j int) { h[i], h[j] = h[j], h[i] }
func (h KVHeap) Len() int { return len(h) }
func (h *KVHeap) Push(x interface{}) {
*h = append(*h, x.(kv))
}
func (h *KVHeap) Pop() interface{} {
old := *h
n := len(old)
x := old[n-1]
*h = old[0 : n-1]
return x
}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.