給出一個二維數組,請將這個二維數組按第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
}