Question

我想按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]。谁能帮我找到问题所在？谢谢。

Answer 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]])
    })

Playground

Answer 2

按定义排序将在切片中移动要排序的项目，因此更改其各自的索引。但是，您并没有移动正在排序的实际值，它们仅在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]])
})

https://play.golang.org/p/6oFBM27bVR-

或者您也可以实现一种类型，以一次对所有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])
}

https://play.golang.org/p/EFUkHWgjo5U

Answer 3

之所以发生这种情况，是因为sort函数在排序时会更改索引，但是伴随的数组v和w保持不变。

执行所需操作的最佳方法是使用结构中包含的v和w创建单个数组，然后对该数组进行排序。

在Go中按v [i] / w [i]对i进行排序的数组

3 个答案: