Leetcode:112.路徑總和
阿新 • • 發佈:2018-11-13
給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。
說明: 葉子節點是指沒有子節點的節點。
示例:
給定如下二叉樹,以及目標和 sum = 22
,
5 / \ 4 8 / / \ 11 13 4 / \ \ 7 2 1
返回 true
, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2
解題思路:
深度優先搜尋。本題需要處理的問題是,從一個結點node開始,尋找當前結點到葉子結點的路徑總和為sum。
1. 當前結點沒有孩子,那麼他就是葉子結點,返回sum==node->val
2. 有孩子必然要往下訪問,可能性有兩種,在左孩子處找到,或者在右孩子處找到。因此return left||right。
class Solution { public: bool hasPathSum(TreeNode* root, int sum) { if (root == NULL) return false; return DFS(root, sum);//從根節點的出發找一個sum-root->val的路徑 } bool DFS(TreeNode* root, int sum) { if (hasNoChild(root)) return sum == root->val; bool left = false, right = false; if (hasLChild(root)) { left=DFS(root->left, sum - root->val); } if (hasRChild(root)) { right = DFS(root->right, sum - root->val); } return left || right; } }; |