刷题的时候可能会遇到只对切片的一部分排序的需求,比如从下标1开始排,如果使用sort.Slice直接取一部分的切片排,可能会发现结果不对。比如这一题,国王游戏,https://www.luogu.com.cn/problem/P1080,虽然难点在高精度,但高精度之前要根据大臣左右手值的乘积进行排序。本题我将国王和大臣保存在一个二维切片pr中,国王是0下标元素,因为只有大臣会参与排序,所以sort的时候是这样做的
sort.Slice(pr[1:], func(i, j int) bool {
return pr[i][2] < pr[j][2]
})
看起来好像没什么问题,结果其实不对。问题就出在sort.Slice(pr[1:], func)这里,第一个参数如果写成部分切片的形式,则自定的sort函数也需要修改下标,如下所示
pr = pr[1:]
sort.Slice(pr[1:], func(i, j int) bool {
return pr[i+1][2] < pr[j+1][2]
})