[数据结构与算法]216. 组合总和 III

Question

216. 组合总和 III

Ideas

思路：遍历组合，满足

1. 元素不重复
2. 和等于n

添加到结果数组中

Code

class Solution:
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        '''
        思路：1-9组合k个数，去重后如果长度不变说明无重复元素，求和等于n返回
        '''
        res = []
        for i in combinations(range(1,10),k):
            if len(i) == len(set(i)) and sum(i) == n:
                res.append(i)
        return res