Go lang排序2D数组

时间:2017-03-06 15:32:55

标签: arrays go

我想在Go中对二维数组进行排序。任何人都可以建议我如何解决这个问题?

例如,如果我有,

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

然后有类似的东西,

sort.Sort(matrix)

2 个答案:

答案 0 :(得分:3)

您必须自己定义如何对此类型进行排序。您可以创建必要的方法来使用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转换为切片并提供适当的较少功能: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)

答案 1 :(得分:2)

考虑一下,您有一个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]
})