我想在Go语言中对二维数组进行排序。请问有人可以提供如何操作的建议吗?
比如说,如果我有一个:
var matrix [3][3]int{
{2,3,1},
{6,3,5},
{1,4,9}
}
那么,是否有类似于这样的东西呢:
sort.Sort(matrix)
你必须自行定义如何对此类型进行排序。 你可以创建必要的方法来使用 sort.Sort
接口,必要时使用指针来改变数组值: https://play.golang.org/p/thdf-k2k3o
type Matrix [3][3]int
func (m Matrix) Len() int { return len(m) }
func (m Matrix) Less(i, j int) bool {
for x := range m[i] {
if m[i][x] == m[j][x] {
continue
}
return m[i][x] < m[j][x]
}
return false
}
func (m *Matrix) Swap(i, j int) { m[i], m[j] = m[j], m[i] }
func main() {
m := Matrix(matrix)
sort.Sort(&m)
}
或者使用 sort.Slice
函数,将 matrix
转换为切片,并提供适当的 less 函数:https://play.golang.org/p/4hrghm9gib
sort.Slice(matrix[:], func(i, j int) bool {
for x := range matrix[i] {
if matrix[i][x] == matrix[j][x] {
continue
}
return matrix[i][x] < matrix[j][x]
}
return false
})
fmt.Println(matrix)
假设你有一个m×n的矩阵,想要按照第k列进行排序
sort.SliceStable(rangearray, func(i, j int) bool {
return rangearray[i][k] < rangearray[j][k]
})
如果你想对m*n矩阵进行排序,且希望按照第k行进行排序
sort.SliceStable(rangearray, func(i, j int) bool {
return rangearray[k][i] < rangearray[k][j]
})