Saturday, August 29, 2015

Leetcode 216. Combination Sum III

https://leetcode.com/problems/combination-sum-iii/

Nothing but DFS.

Solution:
# T:O(k*C(n,k)) S:O(k)
class Solution:
    # @param {integer} k
    # @param {integer} n
    # @return {integer[][]}
    def combinationSum3(self, k, n):
        result = []
        self.combinationSumRecu(result, [], 1, k, n)
        return result
    
    def combinationSumRecu(self, result, intermediate, start, k, target):
        if k == 0 and target == 0:
            result.append(list(intermediate))
        elif k < 0:
            return
        while start < 10 and start * k + k * (k - 1) / 2 <= target:
            intermediate.append(start)
            self.combinationSumRecu(result, intermediate, start + 1, k - 1, target - start)
            intermediate.pop()
            start += 1
Run Time: 44 ms

No comments:

Post a Comment