mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-09 19:44:45 +08:00
feat(1356 数字二进制1排序): 使用Go内置排序函数
This commit is contained in:
@ -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
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
Reference in New Issue
Block a user