LeetCode Combinations

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) 来获得相关源码。