-
Notifications
You must be signed in to change notification settings - Fork 669
/
sorting.go
97 lines (76 loc) · 2.09 KB
/
sorting.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// Copyright (C) 2019-2022, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package utils
import (
"bytes"
"sort"
)
// IsSortedAndUnique returns true if the elements in the data are unique and sorted.
func IsSortedAndUnique(data sort.Interface) bool {
for i := data.Len() - 2; i >= 0; i-- {
if !data.Less(i, i+1) {
return false
}
}
return true
}
type innerSortUint32 []uint32
func (su32 innerSortUint32) Less(i, j int) bool {
return su32[i] < su32[j]
}
func (su32 innerSortUint32) Len() int {
return len(su32)
}
func (su32 innerSortUint32) Swap(i, j int) {
su32[j], su32[i] = su32[i], su32[j]
}
// SortUint32 sorts an uint32 array
func SortUint32(u32 []uint32) {
sort.Sort(innerSortUint32(u32))
}
// IsSortedAndUniqueUint32 returns true if the array of uint32s are sorted and unique
func IsSortedAndUniqueUint32(arr []uint32) bool {
for i := 0; i < len(arr)-1; i++ {
if arr[i] >= arr[i+1] {
return false
}
}
return true
}
type innerSortUint64 []uint64
func (su64 innerSortUint64) Less(i, j int) bool {
return su64[i] < su64[j]
}
func (su64 innerSortUint64) Len() int {
return len(su64)
}
func (su64 innerSortUint64) Swap(i, j int) {
su64[j], su64[i] = su64[i], su64[j]
}
// SortUint64 sorts an uint64 array
func SortUint64(u64 []uint64) {
sort.Sort(innerSortUint64(u64))
}
// IsSortedAndUniqueUint64 returns true if the array of uint64s are sorted and unique
func IsSortedAndUniqueUint64(u64 []uint64) bool {
return IsSortedAndUnique(innerSortUint64(u64))
}
type innerSortBytes [][]byte
func (arr innerSortBytes) Less(i, j int) bool {
return bytes.Compare(arr[i], arr[j]) == -1
}
func (arr innerSortBytes) Len() int {
return len(arr)
}
func (arr innerSortBytes) Swap(i, j int) {
arr[j], arr[i] = arr[i], arr[j]
}
// Sort2DBytes sorts a 2D byte array
// Each byte array is not sorted internally; the byte arrays are sorted relative to another.
func Sort2DBytes(arr [][]byte) {
sort.Sort(innerSortBytes(arr))
}
// IsSorted2DBytes returns true iff [arr] is sorted
func IsSorted2DBytes(arr [][]byte) bool {
return sort.IsSorted(innerSortBytes(arr))
}