[LeetCode By Python] 216. Combination Sum III

一、题目

[LeetCode By Python] 216. Combination Sum III_第1张图片
Combination Sum III

二、解题

输入任意的k,和n,求出k个数加起来数值等于n的所有情况。

这一题由于没接触过回溯法,迟迟不知道怎么下手,观察了网上的一些大神的解法,才知道回溯这个东西,豁然开朗。

三、尝试与结果

class Solution(object):
    def solve(self,k,cur,remainder,prevVal,listT,result):
        if cur == k:
            if remainder > prevVal and remainder <= 9:
                listT.append(remainder)
                one = listT[:]
                result.append(one)
                del listT[-1]
        elif cur < k:
            for i in range(prevVal+1,9-(k-cur)+1):
                listT.append(i)
                self.solve(k, cur+1,remainder-i, i ,listT,result)
                del listT[-1] 

    def combinationSum3(self, k, n):
        result = []
        listT = []
        if k > 0 and k <=9:
            self.solve(k,1,n,0,listT,result)
        return result

结果:AC

理解:
1)传入参数

  • k与cur:两个用来比较判断递归的层次,如果cur当前层已经等于k了,那么就代表是最后一层了,要对result集进行操作了。
  • remainder与prevVal:prevVal代表当前选取了的数字,有两个作用,和remainder比较,用来判断是否需要添加remainder到list(如果remainder < prevVal,即我选了8,剩下了3,,那还怎么选,因为我只有一个9了)。
  • list :list作为回溯的主角,每次尝试append一个值之后,在下层递归后,要删除最后一个元素,保留原始状态(并不关系递归里面的结果,只关系当前状态)

你可能感兴趣的:([LeetCode By Python] 216. Combination Sum III)