update 0077.组合优化: 更换 go 代码

problems/0077.组合优化.md

@@ -133,7 +133,7 @@ public:
 
 # 总结
 
-本篇我们准对求组合问题的回溯法代码做了剪枝优化，这个优化如果不画图的话，其实不好理解，也不好讲清楚。
+本篇我们针对求组合问题的回溯法代码做了剪枝优化，这个优化如果不画图的话，其实不好理解，也不好讲清楚。
 
 所以我依然是把整个回溯过程抽象为一棵树形结构，然后可以直观的看出，剪枝究竟是剪的哪里。
 
@@ -194,28 +194,28 @@ class Solution:
 ```
 Go：
 ```Go
-var res [][]int 
+var (
+    path []int
+    res  [][]int
+)
+
 func combine(n int, k int) [][]int {
-   res=[][]int{}
-   if n <= 0 || k <= 0 || k > n {
-		return res
-	}
-    backtrack(n, k, 1, []int{})
-	return res
+    path, res = make([]int, 0, k), make([][]int, 0)
+    dfs(n, k, 1)
+    return res
 }
-func backtrack(n,k,start int,track []int){
-    if len(track)==k{
-        temp:=make([]int,k)
-        copy(temp,track)
-        res=append(res,temp)
+
+func dfs(n int, k int, start int) {
+    if len(path) == k {  
+        tmp := make([]int, k)
+        copy(tmp, path)
+        res = append(res, tmp)
+        return 
     }
-    if len(track)+n-start+1 < k {
-			return
-		}
-    for i:=start;i<=n;i++{ 
-        track=append(track,i)
-        backtrack(n,k,i+1,track)
-        track=track[:len(track)-1]
+    for i := start; i <= n - (k-len(path)) + 1; i++ {  
+        path = append(path, i)
+        dfs(n, k, i+1)
+        path = path[:len(path)-1]
     }
 }
 ```