题目描述:
题解:
1.按照题目给出的示例,相同数字按照不同顺序的排列组合看作一个。
2.题目中示例1的搜索过程如下:
3.实现思路:
<1>设置一个nums数组,共有n+1个数字,从0-n
<2>设置一个used数组,n+1个数字,初始化为全0,记录nums[i]是否已经被搜索(统一设置为n+1个数字之后,搜索范围从1-n)
<3>res保存最终结果,combination记录当前搜索过程,depth记录当前搜索树深度。
<4>定义一个dfs函数,如果当前depth=k,说明找到一个k个数字的组合,将combination加入res并返回。否则i从idx到n开始搜索,如果对应的used[i]=0,说明对应的nums[i]没有被使用,加入combination,然后进入下一层搜索。下一层从i+1开始避免重复!
<5>回溯,将对应used重新设为0,从combination中删除对应数字。
class Solution(object): def combine(self, n, k): used = [0 for i in range(n+1)] nums = [0] for i in range(1,n+1): nums.append(i) res = [] depth = 0 def dfs(idx,used,depth,combination,res): if depth == k: res.append(combination[:]) return for i in range(idx,n+1): if used[i]==0: used[i]=1 combination.append(nums[i]) depth = depth+1 dfs(i+1,used,depth,combination,res) depth = depth-1 used[i]=0 combination.pop() dfs(1,used,depth,[],res) return res
结果通过,但是执行用时很长。