1. 程式人生 > >Leetcode:112.路徑總和

Leetcode:112.路徑總和

給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。

說明: 葉子節點是指沒有子節點的節點。

示例: 
給定如下二叉樹,以及目標和 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。

                                     

C++程式碼
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;
    }
};