算法基础:数组指定规则排序问题(Golang实现)
发布时间:2020-12-16 18:41:05 所属栏目:大数据 来源:网络整理
导读:给出一个二维数组,请将这个二维数组按第i列(i从1开始)排序,如果第i列相同,则对相同的行按第i+1列的元素排序, 如果第i+1列的元素也相同,则继续比较第i+2列,以此类推,直到最后一列。如果第i列到最后一列都相同,则按原序排列。 样例输入: 1,2,3 2,3,4
给出一个二维数组,请将这个二维数组按第i列(i从1开始)排序,如果第i列相同,则对相同的行按第i+1列的元素排序, 按第2列排序,输出: package huawei
import (
"fmt"
"sort"
)
func Test09Base() {
nums := [][]int{{1, 2, 3},{2, 3, 4}, 1},{1, 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
}
(编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |