劍指 Offer 34. 二叉樹中和為某一值的路徑(中等)
阿新 • • 發佈:2021-08-30
通過率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 };