CERBERUS/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty.git synced 2026-08-04 14:46:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Utility function to sort with key

Browse source
This commit is contained in:
Kovid Goyal 2022-09-21 11:00:40 +05:30
parent e7f38929d9
commit b1e08adbce
No known key found for this signature in database
GPG key ID: 06BC317B515ACE7C
3 changed files with 29 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tools/utils/misc.go
View file

@ -5,6 +5,8 @@ package utils
import (
"fmt"
"sort"
"golang.org/x/exp/constraints"
)
var _ = fmt.Print
@ -33,3 +35,21 @@ func StableSort[T any](s []T, less func(a, b T) bool) []T {
sort.SliceStable(s, func(i, j int) bool { return less(s[i], s[j]) })
return s
}
func SortWithKey[T any, C constraints.Ordered](s []T, key func(a T) C) []T {
mem := make([]C, len(s))
for i, x := range s {
mem[i] = key(x)
}
sort.Slice(s, func(i, j int) bool { return mem[i] < mem[j] })
return s
}
func StableSortWithKey[T any, C constraints.Ordered](s []T, key func(a T) C) []T {
mem := make([]C, len(s))
for i, x := range s {
mem[i] = key(x)
}
sort.SliceStable(s, func(i, j int) bool { return mem[i] < mem[j] })
return s
}