组合

[编程题]组合

热度指数：11807 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 32M，其他语言64M
算法知识视频讲解

给出两个整数n和k，返回从1到n中取k个数字的所有可能的组合

例如：

如果n=4，k=2，结果为

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

示例1

输入

2,1

输出

[[1],[2]]

示例2

输入

3,1

输出

[[1],[2],[3]]

算法知识视频讲解

Python

牛客匿名用户202212

## 二叉树思路，yes or no，是否添加当前index数据
## 递归添加当前路径
def binaryTreeSeq(nums,k,index,path,ans):
        if len(path)==k:
            ans.append(path)
            return 
        if index==len(nums):return
        
        no=path
        binaryTreeSeq(nums,k,index+1,no,ans)
        
        yes=path+[nums[index]]
        binaryTreeSeq(nums,k,index+1,yes,ans)
        
class Solution:
    def combine(self , n , k ):
        nums = list(range(1, n+1))
        ans=[]
        binaryTreeSeq(nums,k,0,[],ans)
        ans=ans[::-1]
        
        return(ans)
        # write code here

发表于 2022-12-19 14:32:58 回复(0)