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