题目描述
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