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

来源:互联网 发布:淘宝买家秀怎么发微淘 编辑:程序博客网 时间:2024/04/30 18:35

递归算法

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

2 0
原创粉丝点击