1. 程式人生 > >leetcode40組合總數 Python

leetcode40組合總數 Python

給定一個數組 candidates 和一個目標數 target ,找出 candidates 中所有可以使數字和為 target 的組合。

candidates 中的每個數字在每個組合中只能使用一次。

說明:

所有數字(包括目標數)都是正整數。
解集不能包含重複的組合。
示例 1:

輸入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集為:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:

輸入: candidates = [2,5,2,1,2], target = 5,
所求解集為:
[
[1,2,2],
[5]
]

思路:看到這道題,馬上可以聯想到上一題。

這兩道題的區別在哪?它們都可以使用陣列內任意個數的元素,但上一題能重複,這一題不能重複。還記得上一次我們是如何製造重複的嗎?我們在遞迴函式中設定了一個start引數,他每次手動遞迴時不加1。這一次不能重複,意思是說我們選了這個元素後就不能再選了,得往後再選一個。

class Solution(object):
    def combinationSum2(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        Solution.ans_list = []
        Solution.target = target
        candidates.sort()
        self.DFS(candidates, target, 0, [])
        return Solution.ans_list
 
    def DFS(self, candidates, target, i, valuelist):
        if target == 0 and valuelist not in Solution.ans_list:
            Solution.ans_list.append(valuelist)
        for i in range(i, len(candidates)):
            if candidates[i] > target:
                return
            #與上一道題唯一的區別就在於下面這個遞迴我們加了1,這樣下次遞迴的迴圈就從i+1開始了
            self.DFS(candidates, target-candidates[i], i+1, valuelist+[candidates[i]])

在這裡插入圖片描述