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