112 二叉樹路徑總和

112 二叉樹路徑總和

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

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

示例:
給定如下二叉樹,以及目標和 sum = 22,

			 5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

返回 true, 因爲存在目標和爲 22 的根節點到葉子節點的路徑 5->4->11->2

解法 - 遞歸

如果不存在根節點,則返回false。

如果存在根節點

  • 左右子樹都爲空,判斷根節點的值是否爲目標值;
  • 當前累加值加上根節點的值;
  • 存在左或者右子樹,遞歸計算累加值,判斷是否滿足目標值。
/**
 * 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) {
        return dfs(root, sum, 0);
    }
    bool dfs(TreeNode* root,int sum,int now){
        if(root==NULL) return false;
        now+=root->val;
        if(!root->left && !root->right && now==sum) return true;
        if(dfs(root->left,sum,now)||dfs(root->right,sum,now)) return true;
        return false;
    }
};

鏈接:https://leetcode-cn.com/problems/path-sum

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章