Go语言如何对一个二维数组进行排序

9

我想在Go语言中对二维数组进行排序。请问有人可以提供如何操作的建议吗?

比如说,如果我有一个:

var matrix [3][3]int{
  {2,3,1},
  {6,3,5},
  {1,4,9}
}

那么,是否有类似于这样的东西呢:

sort.Sort(matrix)

4
你期望输出是什么? - Tyler
2个回答

17

你必须自行定义如何对此类型进行排序。 你可以创建必要的方法来使用 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)

11

假设你有一个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]
})

Go Playground 中有一个 sort.SliceStable 的二维字符串切片 ([][]string) 示例。 - mikequentel

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接