leetcode129. 求根到叶子节点数字之和

传送门

题目:给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。
例如,从根到叶子节点路径 1->2->3 代表数字 123。
计算从根到叶子节点生成的所有数字之和。

输入: [1,2,3]
    1
   / \
  2   3
输出: 25

从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此,数字总和 = 12 + 13 = 25.

1. 回溯, 用列表存每个路径和,太麻烦了:

	public int sumNumbers(TreeNode root) {
        return DFS(root, new LinkedList<>());
    }

    public int DFS(TreeNode root, LinkedList<Integer> path) {
        if (root == null) return 0;
        path.addLast(root.val);
        if (root.left == null && root.right == null) {
            int sum = 0;
            for (int num : path) sum = sum * 10 + num;
            path.removeLast();
            return sum;
        }
        int leftDFSum = DFS(root.left, path);
        int rightDFSum = DFS(root.right, path);
        path.removeLast();// 回溯的是root节点,所以要在左右递归之后弹出
        return leftDFSum + rightDFSum;
	}

2. 优化,不用回溯,直接取节点值就好了,用完就舍弃,存它干吗?!

	public int sumNumbers(TreeNode root) {
        return helper(root, 0);
    }
    
    // helper返回以root为根的ans,sum是访问root之前的路径和
    int helper( TreeNode root, int sum) {
        if (root == null) return 0;
        sum = 10 * sum + root.val;
        if (root.left == null && root.right == null) {
            return sum;
        }
        return helper(root.left, sum) + helper(root.right, sum);
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值