代码随想录算法营Day29 | 39. 组合总和, 40.组合总和II ,131.分割回文串

39. 组合总和

这题主要考在对startIndex的控制上,因为每个元素可以重复拿取,所以在startindex上我们不需要在下一次backtrack的时候进行加1操作,继续使用当前的startIndex即可。

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        res = []
        subset = []
        def backTrack(start,sum):
            if sum == target:
                res.append(subset[:])
            elif sum < target:
                for i in range(start,len(candidates)):
                    subset.append(candidates[i])
                    backTrack(i,sum+candidates[i])
                    subset.pop(-1)
        backTrack(0,0)
        return res

40.组合总和II

这题相对于上题多了一些条件,首先数组里的数字含有重复数字,且每个数字只能使用一次。为了去重,我们先将candidates数组进行排序,而后在backTrack算法中的for循环里增加一个条件就是若当前index大于start则判断他与前一个元素是否相等如果相等则跳过当前循环。具体实现如下

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

131.分割回文串

这题使用回溯加记忆化搜索存储曾经判断过的子字符串是否为回文串的结果。

class Solution:
    def check(self,s):
        left,right = 0,len(s)-1
        while left < right:
            if s[left] != s[right]:
                return False
            left += 1
            right -= 1
        return True
    def partition(self, s: str) -> List[List[str]]:
        subset = []
        res = []
        d = {}
        def backTrack(start,s1):
            if start >= len(s):
                res.append(subset[:])
            else:
                for i in range(start,len(s)):
                    s1 += s[i]
                    if s1 in d:
                        if d[s1]:
                            subset.append(s1)
                            backTrack(i+1,"")
                            subset.pop(-1)
                    else:
                        if (e := self.check(s1)):
                            d[s1] = e
                            subset.append(s1)
                            backTrack(i+1,"")
                            subset.pop(-1)
        backTrack(0,"")
        return res

你可能感兴趣的:(算法,python,leetcode)