二叉树 | 从根到叶节点

问题：路径总和

问题链接

解题思路

采用递归的思路

• root为NULL返回false
• root为叶节点时判断节点值和sum是否相等
• root有孩子时，对有孩子的分支进行判断对结果进行逻辑或运算

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) {
        if(!root) return false;//NULL返回false
        if(!root->left && !root->right) return root->val == sum;//root是根节点 
        //root有孩子节点
        bool flag = false;
        if(root->left) flag = flag || hasPathSum(root->left, sum - root->val);
        if(root->right) flag = flag || hasPathSum(root->right, sum - root->val);
        return flag;
    }
};

问题：求根到叶子节点数字之和

问题链接

解题思路

采用递归的思路，用total记录最后的答案，每次到达叶节点就更新total的值。

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:
    int sumNumbers(TreeNode* root) {
        if(!root) return 0;//如果是空树返回0
        int total = 0;
        solve(root, 0, total);//进行处理
        return total;
    }

    void solve(TreeNode *root, int sum, int &total){
        int tmp = sum * 10 + root->val;//更新数值
        if(!root->left && !root->right){
            total += tmp;//到达叶节点加上这个数
            return;
        } 
        if(root->left) solve(root->left, tmp, total);//有左孩子
        if(root->right) solve(root->right, tmp, total);//有右孩子
    }
};