1. 程式人生 > >216. Combination Sum III 組合總數三

216. Combination Sum III 組合總數三

hat [] desc cas lan script light mil number

Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.


Example 1:

Input: k = 3, n = 7

Output:

[[1,2,4]]


Example 2:

Input: k = 3, n = 9

Output:

[[1,2,6], [1,3,5], [2,3,4]]

Credits:


Special thanks to @mithmatt for adding this problem and creating all test cases.


  1. class Solution:
  2. def combinationSum3(self, k, n):
  3. """
  4. :type k: int
  5. :type n: int
  6. :rtype: List[List[int]]
  7. """
  8. res = []
  9. def gen(l, k, n, cur):
  10. if k == len(l) and
    n == 0:
  11. res.append(l[:])
  12. return
  13. for i in range(cur, 10):
  14. l.append(i)
  15. gen(l, k, n - i, i + 1)
  16. l.pop()
  17. gen([], k, n, 1)
  18. return res



來自為知筆記(Wiz)

216. Combination Sum III 組合總數三