【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],
]
思路
这道题和之前的做的排列组合很相似，一个是数组中所有数字进行组合，这个是规定组合个数并且相同的数字算相同的组合，因此我们可以在排列组合的代码上进行改进。就可以得到答案。详见代码。
解决代码
1 class Solution(object):
2 def combine(self, n, k):
3 """
4 :type n: int
5 :type k: int
6 :rtype: List[List[int]]
7 """
8 list_nums = list(range(1, n+1)) # 先构建一个包含n个数字的数组
9 res = []
10 self.permution(list_nums, k, [], res) # 进行组合
11 return res
12
14
15 def permution(self, nums, k, path, res):
16 if len(path) == k: # 当path中个数等于k的时候，表示得到了一种组合
17 res.append(path)
18 return
19 for i in range(len(nums)): # 从数组第一个元素开始进行组合
20 self.permution(nums[i+1:], k, path+[nums[i]], res)
