216. 组合总和 III

题目

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。

说明：

所有数字都是正整数。
解集不能包含重复的组合。
示例 1:

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

代码

class Solution:
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        candidates = [i for i in range(1, 10)]
        # k = 3
        # target = 9

        candidates = sorted(candidates)
        l = len(candidates)

        res = []

        def backtrace(now, sum, index):

            if sum == n and len(now) == k:
                res.append(now[:])
                return 

            for i in range(index, l):
                if sum + candidates[i] <= n:
                    now.append(candidates[i])
                    backtrace(now, sum+candidates[i], i+1)
                    now.pop()
                else:
                    continue

        backtrace([], 0, 0)
        #print(res)

        ans = set()
        for i in res:
            s = ''
            for j in i:
                s += str(j)
            ans.add(s)
        ans = list(ans)

        #print(ans)


        tmp = []
        i, j = 0, 0
        while(i != len(ans)):
            j = 0
            now_str = ans[i]
            i += 1
            now_list = []
            now_int = ''
            while(j != len(now_str)):
                now_int += now_str[j]
                if int(now_int) not in candidates:
                    j += 1
                else:
                    now_list.append(int(now_int))
                    now_int = ''
                    j += 1
            tmp.append(now_list)
            
        return(tmp)


        
            

结果

执行用时：28 ms, 在所有 Python3 提交中击败了99.21%的用户
内存消耗：13.8 MB, 在所有 Python3 提交中击败了22.92%的用户

总结

在递归+回溯上改条件！！！