LeetCode 77. 组合

给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/combinations
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

class Solution(object):
    def combine(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: List[List[int]]
        """
        if n==0 or k==0:return []
#         回溯法
        def dfs(nums,start,temp):
            if len(temp)==k:
                res.append(temp[:])
                return           
            for i in range(start,len(nums)):
                temp.append(nums[i])
                dfs(nums,i+1,temp)
#                 回溯  下一次dfs递归完成后回溯
                temp.pop()
    
        nums=[i for i in range(1,n+1)]        
        res=[]
        dfs(nums,0,[])
        return res