面試題34 二叉樹中和為某一值的路徑

2021-10-04 05:27:14 字數 2617 閱讀 4112

輸入一棵二叉樹和乙個整數,列印出二叉樹中節點值的和為輸入整數的所有路徑。從樹的根節點開始往下一直到葉節點所經過的節點形成一條路徑。

示例:給定如下二叉樹,以及目標和 sum = 22,

5

/ \4 8

/ / \

11 13 4

/ \ / \

7 2 5 1

返回:

[[5,4,11,2],

[5,8,4,5]

]難點在於如何儲存路徑

法一:可以利用遞迴返回時記錄路徑

# definition for a binary tree node.

# class treenode:

# def __init__(self, x):

# self.val = x

# self.left = none

# self.right = none

class

solution

:def

recursive

(self, root, target)

:if root==

none

:return

none

if root.left==

none

and root.right==

none

:if root.val==target:

return

[[root.val]

]return

none

left=self.recursive(root.left, target-root.val)

right=self.recursive(root.right, target-root.val)

if left!=

none

:for elem in left:

elem.insert(

0, root.val)

if right!=

none

:for elem in right:

elem.insert(

0, root.val)

if left!=

none

and right!=

none

:return left+right

if left!=

none

:return left

if right!=

none

:return right

return

none

defpathsum

(self, root: treenode,

sum:

int)

-> list[list[

int]]:

path=self.recursive(root,

sum)

if path==

none

:return

return path

法二:

在遞迴呼叫時使用乙個全域性棧記錄當前訪問路徑,當葉子節點滿足要求時,將棧內路徑儲存。

# definition for a binary tree node.

# class treenode:

# def __init__(self, x):

# self.val = x

# self.left = none

# self.right = none

class

solution

:def

recursive

(self, root, target)

:if root==

none

:return

if root.left==

none

and root.right==

none

:if root.val==target:

[ x for x in self.stack]

) left=self.recursive(root.left, target-root.val)

right=self.recursive(root.right, target-root.val)

self.stack.pop(

)def

pathsum

(self, root: treenode,

sum:

int)

-> list[list[

int]]:

self.path=

self.stack=

self.recursive(root,

sum)

return self.path

這段**我出了2個bug

一開始把path=和stack=放在類裡第一行,發現leetcode上提交結果和上一次執行結果一樣。而當前結果本應該是。猜測應該是leetcode只執行了我的pathsum函式,類裡面的變數沒有再次初始化。因此我把這兩個變數重新定義在方法裡面。

面試題34 二叉樹中和為某一值的路徑

一 題目 輸入一棵二叉樹和乙個整數,列印出二叉樹中結點值的和為輸入整數的所有路徑。從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。二 關鍵 1.儲存路徑使用棧的後進先出的特點,但是是使用vector實現的 比較好用 2.三 解釋 1 解題思路 當用前序遍歷的方式訪問到某乙個節點時,我們把該...

面試題34 二叉樹中和為某一值的路徑

輸入一顆二叉樹的根節點和乙個整數,列印出二叉樹中結點值的和為輸入整數的所有路徑。路徑定義為從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。考察點 具體例子抽象函式 二叉樹遍歷 思路 如下,遞迴的思想。findpath函式不太好理解。struct treenode void findpath...

面試題34 二叉樹中和為某一值的路徑

題目 輸入一棵二叉樹和乙個整數,列印出二叉樹中結點值的和為輸入整數的所有路徑。從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。include include include include include include using namespace std struct binaryt...