获取二叉树叶子节点个数的递归及非递归算法

2023年3月11日 192次阅读 来源: 递归算法

递归算法

public static<T> int numOfLeavesInRecursion(BinaryTreeNode<T> root){ // 获取二叉树叶子节点的个数的递归算法
        if(root == null)
            return 0;
        if(root.getLeft() == null && root.getRight() == null)
            return 1;
        return numOfLeavesInRecursion(root.getLeft()) + numOfLeavesInRecursion(root.getRight());
    }

非递归算法

将层次遍历稍微修改一下即可：

public static<T> int numOfLeaves(BinaryTreeNode<T> root){ // 获取二叉树叶子节点的个数的非递归算法
        if(root == null)
            return 0;
        DynArrayQueue<BinaryTreeNode<T>> queue = new DynArrayQueue<>();
        queue.enQueue(root);
        int count = 0;
        while(!queue.isEmpty()){
            root = queue.deQueue();
            if(root.getLeft() == null && root.getRight() == null){
                count++;
            } else {
                if(root.getLeft() != null)
                    queue.enQueue(root.getLeft());
                if(root.getRight() != null)
                    queue.enQueue(root.getRight());
            }
        }
        return count;
    }

完整代码可以访问我的GitHub：https://github.com/StriverLi/Data-Structures-and-Algorithms-in-Java/blob/master/src/tree/BinaryTreeNode.java

    原文作者：递归算法
    原文地址: https://blog.csdn.net/StriverLi/article/details/70045640
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞