代码随想录算法训练营第24天|● 理论基础 ● 77. 组合
77. 组合
已解答
中等
相关标签
相关企业
给定两个整数 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 {
tmp, res := make([]int, 0), make([][]int, 0)
dfs(n, k, 1, tmp, &res)
return res
}
func dfs(n int, k int, startindex int, combine []int, res *[][]int) {
if len(combine) == k {
tmp := make([]int, len(combine))
copy(tmp, combine)
*res = append(*res, tmp)
return
}
for i := startindex; i <= n; i++ {
combine = append(combine, i)
dfs(n, k, i+1, combine, res)
combine = combine[:len(combine)-1]
}
return
}