Go 中按 v/w 对 i 数组进行排序

当前位置： > > > > Go 中按 v/w 对 i 数组进行排序

来源：stackoverflow
2024-04-22 19:24:36
0浏览
收藏

有志者，事竟成！如果你在学习Golang，那么本文《Go 中按 v/w 对 i 数组进行排序》，就很适合你！文章讲解的知识点主要包括，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

问题内容

我想按 v[i]/w[i] 降序对索引数组进行排序，其中 v 和 w 是另外两个整数数组。这是我在 go 中尝试过的：

package main
import "fmt"
import "sort"

func main() {
    v := [3]int{5, 6, 3}
    w := [3]int{4, 5, 2}
    indices := make([]int, 3)
    for i := range indices {
        indices[i] = i
    }
    sort.Slice(indices, func(a, b int) bool {
        return float32(v[a])/float32(w[a]) > float32(v[b])/float32(w[b])
    })
    fmt.Println(indices)
}

我期望输出为 [2,0,1] 因为 3/2 > 5/4 > 6/5 但实际输出为 [0,2,1] 。谁能帮我找到问题出在哪里吗？谢谢。

解决方案

为了不改变 v 和 w 数组（这可能会很昂贵），我们可以在 less 函数中添加另一个间接级别

sort.slice(indices, func(a, b int) bool {
        return float32(v[indices[a]])/float32(w[indices[a]]) > float32(v[indices[b]])/float32(w[indices[b]])
    })

按定义排序会移动您在切片中排序的项目，从而更改它们各自的索引。但是，您没有移动要排序的实际值（位于 w 和 v 中），仅移动 indices 切片。

由于 indices 切片包含排序的“索引”，因此您可以使用它来查找实际值进行比较。

sort.slice(indices, func(i, j int) bool {
    return float64(v[indices[i]])/float64(w[indices[i]]) > float64(v[indices[j]])/float64(w[indices[j]])
})

或者您可以实现一个类型来一次对所有 3 个值进行排序，例如：

type indexSorter struct {
    indices, w, v []int
}

func (a indexSorter) Len() int { return len(a.indices) }
func (a indexSorter) Swap(i, j int) {
    a.indices[i], a.indices[j] = a.indices[j], a.indices[i]
    a.w[i], a.w[j] = a.w[j], a.w[i]
    a.v[i], a.v[j] = a.v[j], a.v[i]
}
func (a indexSorter) Less(i, j int) bool {
    return float64(a.v[i])/float64(a.w[i]) > float64(a.v[i])/float64(a.w[j])
}

好了，本文到此结束，带大家了解了《Go 中按 v/w 对 i 数组进行排序》，希望本文对你有所帮助！关注公众号，给大家分享更多Golang知识！