【Leetcode】【python】Combinations

题目大意

求在1到n个数中挑选k个数的所有的组合类型。

解题思路

DFS(回溯法)

代码

直接用数字做

看图，摆明了不让我们这么做么，这最后一个数据集坑死了多少人。

class Solution(object):    def combine(self, n, k):        """        :type n: int        :type k: int        :rtype: List[List[int]]        """        self.result = []        temp = []        self.combineHelper(n, k, 0, 1, temp)        return self.result    def combineHelper(self, n, k, length, num_now, temp):        # print 'start', temp, list_num, length, self.k        if length == k:            # print 'add', temp            self.result.append(temp[:])  # 这里直接写temp，就会导致result内的temp永远是一个实例，后面修改temp会使得result一直在变化            # print 'result', self.result            return        for i in range(num_now, n+1):            temp.append(i)            self.combineHelper(n, k, length+1, i+1, temp)            temp.pop()

转为list做

最后一个测试集TLE

class Solution(object):    def combine(self, n, k):        """        :type n: int        :type k: int        :rtype: List[List[int]]        """        self.k = k        self.result = []        list_num = [i for i in range(1, n+1)]        self.combineHelper(0, [], list_num)        return self.result    def combineHelper(self, length, temp, list_num):        # print 'start', temp, list_num, length, self.k        if length == self.k:            # print 'add', temp            self.result.append(temp[:])  # 这里直接写temp，就会导致result内的temp永远是一个实例，后面修改temp会使得result一直在变化            # print 'result', self.result            return        for i, num in enumerate(list_num):            temp.append(num)            # print temp, list_num[i+1:]            # print '-------'            self.combineHelper(length+1, temp, list_num[i+1:])            temp.pop()

总结

做DFS时，有很多小细节，比如该题，temp需要pop，不能在上面加入result那里做，应该是在调用递归后做。
此题肯定有非递归写法，有空琢磨。