LeetCode题解（python）-77. 组合

LeetCode题解（python）

77. 组合

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

示例:

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

解题心得

本题是组合问题，显然要用递归回溯的方法。

首先找到规律，要从n个元素中取k个元素，可以视为：1. 从n-1个元素中取k个元素；2. 从n-1个元素中取k-1个元素，再加上第n个元素。

这样就从C(n,k)递归回溯到C(n-1,k)和C(n-1,k-1)+Item(n)

执行用时 :52 ms, 在所有 Python3 提交中击败了91.86%的用户

内存消耗 :14.9 MB, 在所有 Python3 提交中击败了10.53%的用户

解题代码

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        if k>n or k==0:
            return []
        if k==1:
            return [[i] for i in range(1,n+1)]
        if k==n:
            return [[i for i in range(1,n+1)]]
        
        leftk = self.combine(n-1,k)
        for item in self.combine(n-1,k-1):
            item.append(n)
            leftk.append(item)
        
        return leftk