[数据结构与算法]求根节点到叶节点数字之和

题目链接

思路

???????再这道题中,二叉树每个节点都对应一个数字,每条从根节点到叶子节点的路径都代表一个数字.其实.每个节点等于其父节点上的数字乘以10再加上该节点的值;根节点上的父节点上的数字为0;只要计算出每个叶子节点上所代表的数字,然后再计算所有叶子节点所代表的数字之和;即可得到结果
???????从根节点开始,遍历每个节点,如果遇到叶子节点,则将叶子节点代表的数字加到数字之和,如果不是叶子节点,则计算其子节点所代表的数字,然后对子节点递归遍历.

java

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode() {
    }

    TreeNode(int val) {
        this.val = val;
    }

    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

class Solution {
    public int sumNumbers(TreeNode root) {
        return sumNumber(root, 0);
    }

    private int sumNumber(TreeNode root, int i) {
        if (root == null) {
            return 0;
        }
        int sum = 10 * i + root.val;
        if (root.left == null && root.right == null){
            return sum;
        }
        return sumNumber(root.left,sum) + sumNumber(root.right,sum);
    }
}

go

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

func sumNumbers(root *TreeNode) int {
	return sumNumber(root, 0)
}

func sumNumber(root *TreeNode, i int) int {
	if root == nil {
		return 0
	}
	sum := 10*i + root.Val
	if root.Right == nil && root.Left == nil {
		return sum
	} else {
		return sumNumber(root.Left, sum) + sumNumber(root.Right, sum)
	}
}