LeetCode每日一題: 路徑總和(No.112)

胖宅老鼠發表於2019-04-05

題目:路徑總和


給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。
說明: 葉子節點是指沒有子節點的節點。
複製程式碼

示例:


給定如下二叉樹,以及目標和 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);
    }
}複製程式碼

相關文章