第二十七天|回溯

今天的题还是回溯,但是每道题都有一点小tricky在里面

39. Combination Sum

这道题的小技巧在于做回溯的时候是从i开始不是从原始的0开始,因为涉及到多个元素出现的问题, 这道题可以和

Way1:

如果总数到了target就把当前组合加入到最后的结果中,如果没有就在for循环中回溯,回溯的index是i并不是之前的216做个对比

class Solution(object):
    def combinationSum(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        res=[]
        path=[]
        candidates.sort()
        def dfs(s,index):
            if s==target:
                res.append(path[:])
                return
            if s>target:
                return
            for i in range(index,len(candidates)):
                path.append(candidates[i])
                dfs(s+candidates[i],i)
                path.pop()        
        dfs(0,0)
        return res

40. Combination Sum II

这道题比上一题有一个判断在里面

Way1:

其实直接在最后是否要把潜在的组合加到结果里面这一步进行重复判断,但是时间复杂度会过高,所以选择在回溯的for循环里面进行判断,这样每次只判断一次当前的遍历的值和i-1的值是不是一样就行了

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        res=[]
        path=[]
        used=set()
        candidates.sort()
        def dfs(s,index):
            if s==target:
                res.append(path[:])
                return
            if s>target:
                return
            for i in range(index,len(candidates)):
                if i!=index and candidates[i] == candidates[i-1]:
                    continue
                path.append(candidates[i])
                dfs(s+candidates[i],i+1)
                path.pop()        
        dfs(0,0)
        return res

131. Palindrome Partitioning

这道题的重点在于要想清楚如何进行字串的划分

Way1:

首先还是想一下判断条件,对于这道题而言,判断条件是是否当前的指针指到了最后一个元素之后..然后就是for循环的部分,一个判断条件就是当前的划分(从index到i) 是不是回文,判断方法就是tmp==tmp[::-1],如果是则进行进一步的回溯

class Solution:
    def partition(self, s: str) -> List[List[str]]:
        res=[]
        path=[]
        def dfs(index):
            if index ==len(s):
                res.append(path[:])
                return
            for i in range(index,len(s)):
                tmp=s[index:i+1]
                if tmp==tmp[::-1]:
                    path.append(tmp)
                    dfs(i+1)
                    path.pop()
        dfs(0)
        return res