1. 程式人生 > 其它 >劍指 Offer 34. 二叉樹中和為某一值的路徑(中等)

劍指 Offer 34. 二叉樹中和為某一值的路徑(中等)

通過率57.8%

題目連結

題目描述:

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

示例:

給定如下二叉樹,以及目標和target = 22,

5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1

返回:

[
[5,4,11,2],
[5,8,4,5]
]

提示:

節點總數 <= 10000

思路:

深搜,有兩個注意點:

  • 不要直接將路徑path追加到res中,否則加入的只是path的記憶體地址,後面path變了res裡所存的路徑也會跟著變,正確做法是先對path進行深拷貝,然後將深拷貝出來的新陣列追加到res中
  • 對於空樹,直接返回空陣列
 1 /*JavaScript*/
 2 /**
 3  * Definition for a binary tree node.
 4  * function TreeNode(val, left, right) {
 5  *     this.val = (val===undefined ? 0 : val)
 6  *     this.left = (left===undefined ? null : left)
 7  *     this.right = (right===undefined ? null : right)
 8  * }
 9  */
10 /*
* 11 * @param {TreeNode} root 12 * @param {number} target 13 * @return {number[][]} 14 */ 15 // 深搜 16 var dfs = function(node, target, path, sum, res) { 17 path.push(node.val) 18 sum += node.val 19 // 葉節點 20 if(!node.left && !node.right) { 21 if(sum === target) { 22 //
對path進行深拷貝並追加到res中 23 const temp = [] 24 path.forEach(item => { 25 temp.push(item) 26 }) 27 res.push(temp) 28 } 29 return path.pop() 30 } 31 if(node.left) dfs(node.left, target, path, sum, res) 32 if(node.right) dfs(node.right, target, path, sum, res) 33 path.pop(node.val) 34 } 35 36 var pathSum = function(root, target) { 37 if(!root) return [] 38 const res = [] 39 dfs(root, target, [], 0, res) 40 return res 41 };