題目:路徑總和
給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。
說明: 葉子節點是指沒有子節點的節點。
複製程式碼
示例:
給定如下二叉樹,以及目標和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ \
7 2 1
返回 true, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2。
複製程式碼
思考:
利用遞迴,每次將sum與當前節點的val相減。
如果當前節點為葉子節點,則判斷sum減去當前節點的val是否等於0,為0則說明路徑滿足要求。
如非葉子節點,則遞迴呼叫自身傳入當前節點的左右子節點和sum-root.val的值。
複製程式碼
實現:
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
//當前節點為null返回false
if (root == null) {
return false;
}
//左右子節點都為null說明是葉子節點,則計算sun減去當前節點的val如果為0則說明這條路近滿足要求
if (root.left == null && root.right == null) {
return sum - root.val == 0;
}
//若非葉子節點進行遞迴呼叫,將sum值接減去當前節點val之後的值與當前節點的子節點傳入
// 因為左右有一條路徑滿足要求即可,所以用||或連線
return hasPathSum(root.left, sum - root.val) || hasPathSum(root.right, sum - root.val);
}
}複製程式碼