【LeetCode每天一题】Combinations(组合)

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

Example:

Input: n = 4, k = 2
Output:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

思路

---

　　这道题和之前的做的排列组合很相似，一个是数组中所有数字进行组合，这个是规定组合个数并且相同的数字算相同的组合，因此我们可以在排列组合的代码上进行改进。就可以得到答案。详见代码。

解决代码

---

 class Solution(object):
     def combine(self, n, k):
         """
         :type n: int
         :type k: int
         :rtype: List[List[int]]
         """
         list_nums = list(range(1, n+1))  # 先构建一个包含n个数字的数组
         res = []
         self.permution(list_nums, k, [], res)  # 进行组合
         return res

     def permution(self, nums, k, path, res):
         if len(path) == k:        # 当path中个数等于k的时候，表示得到了一种组合
             res.append(path)
             return
         for i in range(len(nums)):  # 从数组第一个元素开始进行组合
             self.permution(nums[i+1:], k, path+[nums[i]], res)