Golang算法问题之数组按指定规则排序的方法分析
程序员文章站
2022-06-24 09:22:35
本文实例讲述了golang算法问题之数组按指定规则排序的方法。分享给大家供大家参考,具体如下:
给出一个二维数组,请将这个二维数组按第i列(i从1开始)排序,如果第i列相...
本文实例讲述了golang算法问题之数组按指定规则排序的方法。分享给大家供大家参考,具体如下:
给出一个二维数组,请将这个二维数组按第i列(i从1开始)排序,如果第i列相同,则对相同的行按第i+1列的元素排序,
如果第i+1列的元素也相同,则继续比较第i+2列,以此类推,直到最后一列。如果第i列到最后一列都相同,则按原序排列。
样例输入:
1,2,3
2,3,4
2,3,1
1,3,1
按第2列排序,输出:
1,2,3
2,3,1
1,3,1
2,3,4
代码实现:
复制代码 代码如下:
package huawei
import (
"fmt"
"sort"
)
func test09base() {
nums := [][]int{{1, 2, 3}, {2, 3, 4}, {2, 3, 1}, {1, 3, 1}}
firstindex := 2 //按第二列排序
result := arraysort(nums, firstindex-1)
fmt.println(result)
}
//按指定规则对nums进行排序(注:此firstindex从0开始)
func arraysort(nums [][]int, firstindex int) [][]int {
//检查
if len(nums) <= 1 {
return nums
}
if firstindex < 0 || firstindex > len(nums[0])-1 {
fmt.println("warning: param firstindex should between 0 and len(nums)-1. the original array is returned.")
return nums
}
//排序
mintarray := &intarray{nums, firstindex}
sort.sort(mintarray)
return mintarray.marr
}
type intarray struct {
marr [][]int
firstindex int
}
//intarray实现sort.interface接口
func (arr *intarray) len() int {
return len(arr.marr)
}
func (arr *intarray) swap(i, j int) {
arr.marr[i], arr.marr[j] = arr.marr[j], arr.marr[i]
}
func (arr *intarray) less(i, j int) bool {
arr1 := arr.marr[i]
arr2 := arr.marr[j]
for index := arr.firstindex; index < len(arr1); index++ {
if arr1[index] < arr2[index] {
return true
} else if arr1[index] > arr2[index] {
return false
}
}
return i < j
}
import (
"fmt"
"sort"
)
func test09base() {
nums := [][]int{{1, 2, 3}, {2, 3, 4}, {2, 3, 1}, {1, 3, 1}}
firstindex := 2 //按第二列排序
result := arraysort(nums, firstindex-1)
fmt.println(result)
}
//按指定规则对nums进行排序(注:此firstindex从0开始)
func arraysort(nums [][]int, firstindex int) [][]int {
//检查
if len(nums) <= 1 {
return nums
}
if firstindex < 0 || firstindex > len(nums[0])-1 {
fmt.println("warning: param firstindex should between 0 and len(nums)-1. the original array is returned.")
return nums
}
//排序
mintarray := &intarray{nums, firstindex}
sort.sort(mintarray)
return mintarray.marr
}
type intarray struct {
marr [][]int
firstindex int
}
//intarray实现sort.interface接口
func (arr *intarray) len() int {
return len(arr.marr)
}
func (arr *intarray) swap(i, j int) {
arr.marr[i], arr.marr[j] = arr.marr[j], arr.marr[i]
}
func (arr *intarray) less(i, j int) bool {
arr1 := arr.marr[i]
arr2 := arr.marr[j]
for index := arr.firstindex; index < len(arr1); index++ {
if arr1[index] < arr2[index] {
return true
} else if arr1[index] > arr2[index] {
return false
}
}
return i < j
}
希望本文所述对大家go语言程序设计有所帮助。
下一篇: python 数据类型 常用法方