leetcode第216题,给定数字和加和个数,要求输出所有可能的组合,并且不能有重复。
典型的回溯问题,如果回溯过程处理得好,就可以顺理成章地去重。
基本思路是,写一个深搜函数,递归的出口条件是加和个数已经达到指定个数并且和为指定的数字,则拷贝结果列表,加入最终的结果当中。如果递归层数已经超过指定加和数目,则立刻返回,剩余的情况就是不断地遍历1-9所有数字,并不断填入结果列表,并进行深度搜索,这里可以稍微进行一下剪枝操作,每次不必从1开始搜索,而是从原始的结果列表的后一个数开始搜,这样可以去重,另外加和时如果遇到数字已经比目标和还大,那么可以直接跳出循环。
最后为了能保证第一次顺利执行深搜,列表中初始化为0,不影响最终加和,在最终填入最终结果列表时去掉0即可。
class Solution(object):
def combinationSum3(self, k, n):
"""
:type k: int
:type n: int
:rtype: List[List[int]]
"""
import copy
ans = []
def dfs(sumList, step):
if step == k and sum(sumList) == n:
tmp = copy.deepcopy(sumList)
del tmp[0]
ans.append(tmp)
elif step < k:
for i in range((sumList[-1]+1),10):
if i >= n:
break
else:
sumList.append(i)
dfs(sumList, step+1)
del sumList[-1]
else:
return
dfs([0],0)
return ans