简单算法 二叉树根节点到叶子节点的所有路径和(java)

简单算法 二叉树根节点到叶子节点的所有路径和(java)
描述
给定一个仅包含数字\ 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

public int sumNumbers (TreeNode root) {
        if(root == null) return 0;
        Stack<TreeNode> s1 = new Stack<TreeNode>();
        Stack<Integer> s2 = new Stack<Integer>();
        int sum = 0;
        s1.push(root);
        s2.push(root.val);
        while(!s1.isEmpty()){
            TreeNode node = s1.pop();
            int value = s2.pop();
            if(node.left == null && node.right == null){
                sum = sum + value;
            }else{
                if(node.right != null) {
                    s1.push(node.right);
                    s2.push(value * 10 + node.right.val);
                }
                if(node.left != null) {
                    s1.push(node.left);
                    s2.push(value * 10 + node.left.val);
                }
            }
        }
        return sum;
    }

递归解法:

public int sumNumbers (TreeNode root) {
        ArrayList<Integer> list = new ArrayList<>();
        test(root, 0 , list);
        int sum = 0;
        for(int i = 0; i < list.size(); i++){
            sum = sum + list.get(i);
        }
        return sum;
    }
    public void test(TreeNode root,int value,ArrayList<Integer> list){
        if(root == null) return;
        value = root.val + 10 * value; 
        if(root.left == null && root.right == null){
            list.add(value);
        }
        test(root.left, value, list);
        test(root.right, value, list);
    }
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值