KK's blog

每天积累多一些

0%

LeetCode 216 Combination Sum III

LeetCode

<div>

Find all valid combinations of k numbers that sum up to n such that the following conditions are true:

  • Only numbers 1 through 9 are used.
  • Each number is used at most once.

Return a list of all possible valid combinations. The list must not contain the same combination twice, and the combinations may be returned in any order.

Example 1:

<pre>Input: k = 3, n = 7 Output: [[1,2,4]] Explanation: 1 + 2 + 4 = 7 There are no other valid combinations.</pre>

Example 2:

<pre>Input: k = 3, n = 9 Output: [[1,2,6],[1,3,5],[2,3,4]] Explanation: 1 + 2 + 6 = 9 1 + 3 + 5 = 9 2 + 3 + 4 = 9 There are no other valid combinations. </pre>

Example 3:

<pre>Input: k = 4, n = 1 Output: [] Explanation: There are no valid combinations. Using 4 different numbers in the range [1,9], the smallest sum we can get is 1+2+3+4 = 10 and since 10 > 1, there are no valid combination. </pre>

Constraints:

  • 2 <= k <= 9
  • 1 <= n <= 60

</div>

题目大意:

数字1-9的组合个数为k的组合和等于k,每个元素最多用一次

解题思路:

用组合模板,先排序

解题步骤:

N/A

注意事项:

  1. Leetcode 40和77的结合。个数和target都要达到。用if k == 0 and target == 0

Python代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
def combinationSum3(self, k: int, n: int) -> List[List[int]]:
nums = [_ for _ in range(1, 10)]
res = []
self.dfs(nums, 0, k, n, [], res)
return res

def dfs(self, nums, start, k, target, path, res):
if k == 0 and target == 0:
res.append(list(path))
return
if k == 0:
return
for i in range(start, len(nums)):
path.append(nums[i])
self.dfs(nums, i + 1, k - 1, target - nums[i], path, res)
path.pop()

算法分析:

时间复杂度为<code>O(2<sup>k</sup>)</code>,空间复杂度O(n)

Free mock interview