【力扣40. 组合总和 II】回溯(python3)

题目描述

https://leetcode-cn.com/problems/combination-sum-ii/

思路题解

一开始的错误做法:
会卡在相同的小的数字特别多的情况:
[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1]
27
[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1]
30

因为剪枝不当,导致会出现下方这种情况
1-2-5
 -2-5
 -2-5

1-1-1
 -1-1
 -1-1
#错误方法:时间超限
class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates.sort()
        self.ans=[]
        def recall(res,count,i,n):
            if i>=n:return 
            count+=candidates[i]
            if count==target:
                if res+[candidates[i]] not in self.ans:self.ans.append(res+[candidates[i]])
            elif count<target:
                for j in range(i+1,n):
                    recall(res+[candidates[i]],count,j,n)
        recall([],0,0,len(candidates))
        return self.ans

遂换正确方法,且注意每次回溯的时候当前的数字是不算在结果里的,只判断之前的结果是否正确。要先排序,不然还要判断一下重复。
count代表已有

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates.sort()
        self.ans=[]
        def recall(res,count,i):#i是没有便利过的,count代表已有
            if count==target:
                self.ans.append(res)
            for j in range(i,n):
                if count+candidates[j]>target:break
                #下方杜绝了1-2-5 的情况 但可以出现1-2-2的情况
                #          -2-5
                if j>i and candidates[j-1]==candidates[j]:continue
                recall(res+[candidates[j]],count+candidates[j],j+1)
        i,n=0,len(candidates)
        recall([],0,i)
        return self.ans

在这里插入图片描述

优化一下count的计算方式,count代表剩余

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates.sort()
        self.ans=[]
        def recall(res,count,i):#i是没有便利过的,count代表剩余
            if count==0:
                self.ans.append(res)
            for j in range(i,n):
                if candidates[j]>count:break
                #下方杜绝了1-2-5 的情况 但可以出现1-2-2的情况
                #          -2-5
                if j>i and candidates[j-1]==candidates[j]:continue
                recall(res+[candidates[j]],count-candidates[j],j+1)
        i,n=0,len(candidates)
        recall([],target,i)
        return self.ans

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值