二叉树中等 牛客NC.5二叉树根节点到叶子节点的所有路径和

二叉树根节点到叶子节点的所有路径和

题目描述

给定一个仅包含数字\ 0-9 0−9 的二叉树，每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1\to 2\to 31→2→3,那么这条路径就用\ 123 123 来代替。
找出根节点到叶子节点的所有路径表示的数字之和
例如：

这颗二叉树一共有两条路径，
根节点到叶子节点的路径 1\to 21→2 用数字\ 12 12 代替
根节点到叶子节点的路径 1\to 31→3 用数字\ 13 13 代替
所以答案为\ 12+13=25 12+13=25

思路：

• 先序遍历，当到达叶子结点（说明遍历完一条路径）便返回这一条路径数。当到达的结点是空（说明该结点的父节不是叶子结点，要不然在父节点哪里就返回了，因此不构成路径）返回0。
• 向下递归需要传送前面结点已经组成的数。
• 每次递归向上返回的是通过该结点的所有路径数之和。（对于根结点来说，经过左孩子的所有路径数之和与经过右孩子的所有路径数之和的总和便是整棵树的路径数之和）

代码：

public class Solution {
    public int sumNumbers (TreeNode root) {
        return count(root,0);
    }
    public int count(TreeNode root, int sum){
        if(root == null){
            return 0;
        }
        sum = sum*10 + root.val;
        if(root.left == null && root.right == null){//若是叶子结点，返回sum
            return sum;
        }
        return count(root.left,sum)+count(root.right,sum);
    }
}