Merge pull request #2289 from aFlyBird0/batch-1

改写两处Go代码，使其更符合Go习惯

problems/0213.打家劫舍II.md

@@ -225,32 +225,38 @@ class Solution:
 // 打家劫舍Ⅱ 动态规划
 // 时间复杂度O(n) 空间复杂度O(n)
 func rob(nums []int) int {
-    if len(nums) == 1 {
+    // 如果长度为0或1，那么有没有环的限制都一样
-        return nums[0]
+    if len(nums) <= 1 {
+        return robWithoutCircle(nums)
     }
-    if len(nums) == 2 {
+
-        return max(nums[0], nums[1])
+    // 否则，去头或去尾，取最大
-    }
+    res1 := robWithoutCircle(nums[:len(nums)-1])
-    
+    res2 := robWithoutCircle(nums[1:])
-    result1 := robRange(nums, 0)
+
-    result2 := robRange(nums, 1)
+    return max(res1, res2)
-    return max(result1, result2)
 }
 
-// 偷盗指定的范围
+// 原始的打家劫舍版
-func robRange(nums []int, start int) int {
+func robWithoutCircle(nums []int) int {
+    switch len(nums) {
+        case 0: return 0
+        case 1: return nums[0]
+    }
     dp := make([]int, len(nums))
-    dp[1] = nums[start]
+    dp[0]=nums[0]
-    
+    dp[1] = max(nums[0], nums[1])
-    for i := 2; i < len(nums); i++ {
+
-        dp[i] = max(dp[i - 2] + nums[i - 1 + start], dp[i - 1])
+    for i:=2; i<len(nums); i++ {
+        dp[i] = max(dp[i-1], dp[i-2]+nums[i])
     }
-    
+
-    return dp[len(nums) - 1]
+    return dp[len(nums)-1]
+
 }
 
-func max(a, b int) int {
+func max(a, b int ) int {
-    if a > b {
+    if a>b {
         return a
     }
     return b

problems/1356.根据数字二进制下1的数目排序.md

@@ -171,36 +171,29 @@ class Solution:
 
 ```go
 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
 }
 
-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++
     }
+
     return count
 }
 ```