代码随想录算法营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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值