给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
示例 2:
输入:n = 1, k = 1 输出:[[1]]
提示:
1 <= n <= 20
1 <= k <= n
题解:
回溯法:
func combine(n int, k int) [][]int {
var path []int
var result [][]int
var backtracking func(n, k, startIndex int)
backtracking = func(n, k, startIndex int) {
if len(path) == k {
tmp := make([]int, k)
copy(tmp, path)
result = append(result, tmp)
return
}
for i := startIndex; i <= n; i++ {
path = append(path, i)
backtracking(n, k, i+1)
path = path[:len(path)-1]
}
}
backtracking(n, k, 1)
return result
}
回溯法(剪枝优化):
func combine(n int, k int) [][]int {
var path []int
var result [][]int
var backtracking func(n, k, startIndex int)
backtracking = func(n, k, startIndex int) {
if len(path) == k {
tmp := make([]int, k)
copy(tmp, path)
result = append(result, tmp)
return
}
for i := startIndex; i <= n-(k-len(path))+1; i++ {
path = append(path, i)
backtracking(n, k, i+1)
path = path[:len(path)-1]
}
}
backtracking(n, k, 1)
return result
}
评论前必须登录!
注册