当前位置: 代码迷 >> 综合 >> 左叶子之和(BFS)
  详细解决方案

左叶子之和(BFS)

热度:91   发布时间:2024-02-22 00:08:03.0

计算给定二叉树的所有左叶子之和.

示例:

                        6
                    /      \
                4             8
            /     \          /     \
         1         5      7         9
                                      /
                                  11

在这个二叉树中,有两个左叶子,分别是 1、 7和11,所以返回 19

一个节点为左叶子节点,当且仅当它是某个节点的左子节点,并且它是一个叶子结点。因此我们可以考虑对整棵树进行遍历,当我们遍历到节点时,如果它的左子节点是一个叶子结点,那么就将它的左子节点的值累加计入答案。

package com.loo;

import java.util.Stack;

public class StackLeafSum {
    
    static class TreeNode {
        TreeNode left;
        TreeNode right;
        int value;
        TreeNode(int v) {
            value = v;
        }
    }


    public static void main(String[] args) {

        // 构造示例的二叉树
        TreeNode root = new TreeNode(6);
        TreeNode l1 = new TreeNode(4);
        TreeNode l2 = new TreeNode(1);
        TreeNode l3 = new TreeNode(5);
        TreeNode r1 = new TreeNode(8);
        TreeNode r2 = new TreeNode(7);
        TreeNode r3 = new TreeNode(9);
        TreeNode r4 = new TreeNode(11);
        root.left = l1;
        l1.left = l2;
        l1.right = l3;
        root.right = r1;
        r1.left = r2;
        r1.right = r3;
        r3.left = r4;
        System.out.println(stackLeafSum(root));
    }
    
    public static int stackLeafSum(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int sum = 0;
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.clear();
        stack.push(root);
        while (stack.size()>0) {
            TreeNode node = stack.pop();
            if (node.left != null) {
                if (isLeafNode(node.left)) {
                    sum += node.left.value;
                } else {
                    stack.push(node.left);
                }
            }
            if (node.right != null) {
                stack.push(node.right);
            }
        }
        return sum;
    }
    

    // 判断是否是叶子节点
    public static boolean isLeafNode(TreeNode node) {
        return node.left == null && node.right == null;
    }

}

 

  相关解决方案