leetcode 面試題216 組合總和 III

2021-10-04 04:07:59 字數 1317 閱讀 7014

2020/3/20  打卡

找出所有相加之和為 n 的 k 個數的組合。組合中只允許含有 1 - 9 的正整數,並且每種組合中不存在重複的數字。

說明:所有數字都是正整數。

解集不能包含重複的組合。

示例 1:

輸入: k = 3, n = 7

輸出: [[1,2,4]]

示例 2:

輸入: k = 3, n = 9

輸出: [[1,2,6], [1,3,5], [2,3,4]]

一如既往的 dfs套路吧,進行回溯+剪枝。

1.初試化結果陣列res

2.定義回溯函式help(count,i,tmp,target),其中count表示當前已經使用的數字數,i表示當前訪問的數字,tmp表示當前中間結果,target表示下一步的目標和。

(1)若count==k,說明已經使用了k個數:若target==0,表示tmp的和等於n,將tmp加入res。return

(2)遍歷區間[i,10)[i,10):

*剪枝,若j>target,說明接下來的數字都比目標和大,break

*執行回溯help(count+1,j+1,tmp+[j],target−j)

3.執行help(0,1,,n)

4.返回res

# 使用 dfs+剪枝的操作,對 數字數量進行限制。

# 時間複雜度為o(n!) 空間複雜度為o(k)

class solution(object):

def combinationsum3(self, k, n):

""":type k: int

:type n: int

:rtype: list[list[int]]

"""res =

# 內部dfs搜尋 (count表示當前已經使用的數字數量,i表示當前訪問的索引,tmp表示當前中間結果,target表示下一步的目標和)

def dfs(count, i, tmp, target):

# dfs截止的 邊界條件 ,先是數量滿足,再是和要對應

if count == k:

if target == 0:

for j in range(i, 10):

# 暴力之下的剪枝設定。

if (j > target):

break

# 啟動內部深搜尋。 j+1是分支位置

dfs(count + 1, j + 1, tmp + [j], target - j)

# 外部啟動dfs搜尋

dfs(0, 1, , n)

return res

leetcode 面試題專題

面試題51.陣列中的逆序對 在陣列中的兩個數字,如果前面乙個數字大於後面的數字,則這兩個數字組成乙個逆序對。輸入乙個陣列,求出這個陣列中的逆序對的總數。利用歸併排序,將陣列分成兩部分,前面一部分和後面一部分,進行歸併的時候,需要比較兩個陣列的數值大小,如果第二個陣列的值小的話,代表前乙個陣列剩下的都...

位元組面試題 leetcode

給定乙個迴圈陣列 最後乙個元素的下乙個元素是陣列的第乙個元素 輸出每個元素的下乙個更大元素。數字 x 的下乙個更大的元素是按陣列遍歷順序,這個數字之後的第乙個比它更大的數,這意味著你應該迴圈地搜尋它的下乙個更大的數。如果不存在,則輸出 1。示例 1 輸入 1,2,1 輸出 2,1,2 解釋 第乙個 ...

leetcode 面試題40 組合總和 II

2020 3 19 打卡 給定乙個陣列 candidates 和乙個目標數 target 找出 candidates 中所有可以使數字和為 target 的組合。candidates 中的每個數字在每個組合中只能使用一次。說明 所有數字 包括目標數 都是正整數。解集不能包含重複的組合。示例 1 輸入...