代码随想录算法训练营第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
}