feat(1356 数字二进制1排序): 使用Go内置排序函数

This commit is contained in:
Bird
2023-09-28 22:13:00 +08:00
parent 5117a28842
commit 546d3b601a

View File

@ -171,36 +171,29 @@ class Solution:
```go ```go
func sortByBits(arr []int) []int { func sortByBits(arr []int) []int {
var tmp int // 是否arr[i]<=arr[j]
for i := 0; i < len(arr); i++ { // 先比较1的数量后比较值本身
for j := i+1; j < len(arr); j++ { cmp := func(i, j int) bool {
// 冒泡排序的手法但是排序的规则从比大小变成了比位运算1的个数 c1, c2 := bitCount(arr[i]), bitCount(arr[j])
if isCmp(arr[i], arr[j]) { if c1 == c2 {
tmp = arr[i] return arr[i] <= arr[j]
arr[i] = arr[j]
arr[j] = tmp
}
} }
return c1 <= c2
} }
// 调用库函数
// 第一个参数是待排序切片第二个是第i位是否小于第j位的函数
sort.Slice(arr, cmp)
return arr return arr
} }
func isCmp(a, b int) bool { func bitCount(num int) (count int) {
bitA := bitCount(a) for num != 0 {
bitB := bitCount(b) num &= num-1 // 每次运算将最右侧的1变成0
if bitA == bitB {
return a > b
} else {
return bitA > bitB
}
}
func bitCount(n int) int {
count := 0
for n != 0 {
n &= (n-1) // 清除最低位的1
count++ count++
} }
return count return count
} }
``` ```