[Leetcode/파이썬] 39. Combination Sum

Combination Sum

---

Given an array of distinct integers candidates and a target integer target, return a list of all unique combinations of candidates where the chosen numbers sum to target. You may return the combinations in any order.

The same number may be chosen from candidates an unlimited number of times. Two combinations are unique if the frequency of at least one of the chosen numbers is different.

The test cases are generated such that the number of unique combinations that sum up to target is less than 150 combinations for the given input.

 

Example 1:

Input: candidates = [2,3,6,7], target = 7
Output: [[2,2,3],[7]]
Explanation:
2 and 3 are candidates, and 2 + 2 + 3 = 7. Note that 2 can be used multiple times.
7 is a candidate, and 7 = 7.
These are the only two combinations.

Example 2:

Input: candidates = [2,3,5], target = 8
Output: [[2,2,2,2],[2,3,3],[3,5]]

Example 3:

Input: candidates = [2], target = 1
Output: []

 

Constraints:

• 1 <= candidates.length <= 30
• 2 <= candidates[i] <= 40
• All elements of candidates are distinct.
• 1 <= target <= 40

 

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        res = []
        self.dfs(candidates, target, 0, [], res)
        return res


    def dfs(self, candidates, target, idx, path, res) :
        if target < 0 :
            return
        elif target == 0 :
            res.append(path[:])
            return 

        for i in range(idx, len(candidates)) :
            self.dfs(candidates, target - candidates[i], i, path + [candidates[i]], res)

전에 분명 풀었던 문제 같은데 지금 보니 또 새롭네... 콤비네이션 하면 떠오르는 알고리즘이 바로 백트래킹 알고리즘이다. 계산을 다음과 같이 하자. 현재의 값(target)을 두고 리스트(candidates)에 있는 수를 하나씩 뺀다. 그리고 그 수를 답안 리스트(path)에 추가한다.

만약 현재 값이 0보다 작으면, 답안 리스트에 담긴 값들의 합이 정답(target)보다 크기 때문에 가장 마지막에 담았던 수를 제거 한다. 그리고 현재 값이 0이면 이 답안 리스트는 정답이기 때문에 res에 추가하고 직전 상태로 돌아간다. 이 스택 구조가 머리에 안들어오네.