LeetCode演算法題112:路徑總和解析
阿新 • • 發佈:2018-12-19
給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。 說明: 葉子節點是指沒有子節點的節點。
示例: 給定如下二叉樹,以及目標和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ \
7 2 1
返回 true, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2。
因為遞迴時傳遞的是sum值,對於樹來說都是從上往下遍歷,所以遞迴的時候應該使用減法,每次遞迴這個sum值都會減去之前遍歷過的節點的值,到最後的葉子節點如果剛好等於減完的值,那就說明是相等的。如果不動sum值增加一個函式的話,其實可以再新增一個add值來記錄前面所有節點的和,然後最後到了葉子節點再去進行對比,但是這樣就稍微複雜了一點。C++程式使用了這種稍微複雜了一點的方法,python程式使用了較為簡單的方法。 C++原始碼:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
int add = 0;
return isEqual(root, sum, add);
}
bool isEqual(TreeNode* root, int sum, int add) {
if (root==NULL) return false;
if (root->left==NULL && root->right==NULL) return (sum==(add+root->val));
return isEqual(root->left, sum, add+root->val) || isEqual(root-> right, sum, add+root->val);
}
};
python3原始碼:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def hasPathSum(self, root, sum):
"""
:type root: TreeNode
:type sum: int
:rtype: bool
"""
if root == None:
return False
if root.left==None and root.right==None:
return root.val == sum
return self.hasPathSum(root.left, sum-root.val) or self.hasPathSum(root.right, sum-root.val)