LeetCode解题之Combinations
原题
求在1到n个数中挑选k个数的所有的组合类型。
注意点:
- 每个数字只能够用一遍
- 组合的排列没有顺序要求
例子:
输入: n = 4, k = 2
输出: [[1, 4], [2, 4], [3, 4], [1, 3], [2, 3], [1, 2]]
解题思路
采用递归的方式,在n个数中选k个,如果n大于k,那么可以分类讨论,如果选了n,那么就是在1到(n-1)中选(k-1)个,否则就是在1到(n-1)中选k个。递归终止的条件是k为1,这时候1到n都符合要求。
AC源码
class Solution(object):
def combine(self, n, k):
"""
:type n: int
:type k: int
:rtype: List[List[int]]
"""
if k == 1:
return [[i + 1] for i in range(n)]
result = []
if n > k:
result = [r + [n] for r in self.combine(n - 1, k - 1)] + self.combine(n - 1, k)
else:
result = [r + [n] for r in self.combine(n - 1, k - 1)]
return result
if __name__ == "__main__":
assert Solution().combine(4, 2) == [[1, 4], [2, 4], [3, 4], [1, 3], [2, 3], [1, 2]]
欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。