leetcode.cn_求根节点到叶节点数字之和_129

在这里插入图片描述
在这里插入图片描述

public int sumNumbers(TreeNode root){

        /**
         *
         * 使用广度优先搜索,需要维护两个队列,分别存储节点和节点对应的数字。
         * 初始时,将根节点和根节点的值分别加入两个队列。每次从两个队列分别取出一个节点和一个数字,进行如下操作:
         * 如果当前节点是叶子节点,则将该节点对应的数字加到数字之和;
         * 如果当前节点不是叶子节点,则获得当前节点的非空子节点,
         * 并根据当前节点对应的数字和子节点的值计算子节点对应的数字,然后将子节点和子节点对应的数字分别加入两个队列。
         * 搜索结束后,即可得到所有叶子节点对应的数字之和。
         */
        if ( root == null){
            return 0;
        }

        int sum = 0;

        // 存储node
        Queue<TreeNode> nodeQueue = new LinkedList<TreeNode>();
        // 存储node.val
        Queue<Integer> numQueue = new LinkedList<Integer>();
        // 初始时,将根节点和根节点的值分别加入两个队列。每次从两个队列分别取出一个节点和一个数字
        nodeQueue.offer(root);
        numQueue.offer(root.val);
        //
        while( !nodeQueue.isEmpty()){
            TreeNode node = nodeQueue.poll();
            int num = numQueue.poll();
            TreeNode left = node.left ,right = node.right;
            // 如果当前节点是叶子节点,则将该节点对应的数字加到数字之和;
            if (left == null && right == null){
                sum += num;
            } else {
                //如果当前节点不是叶子节点,则获得当前节点的非空子节点,
                // 并根据当前节点对应的数字和子节点的值计算子节点对应的数字,然后将子节点和子节点对应的数字分别加入两个队列。
                if (left != null){
                    nodeQueue.offer(left);
                    numQueue.offer(num * 10 + left.val);
                }
                if (right != null){
                    nodeQueue.offer(right);
                    numQueue.offer(num * 10 + right.val);
                }
            }
        }
        return sum;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值