Today's Challenge

https://leetcode.com/problems/combination-sum-iii/

Combination Sum III - LeetCode

itertools 패키지의 combinations을 사용하면 매우 간단히 풀 수 있다.
다른 사람이 푼 걸 보니 backtraking으로도 풀 수 있는 것 같다.

from itertools import combinations
class Solution:
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        ar = [i for i in range(1,10)]
        res = []
        for c in combinations(ar, k):
            if sum(c) == n:
                res.append(c)
        return res