求二叉树深度的递归和非递归算法

2023年5月28日 173次阅读 来源: 递归算法

递归算法

public static<T> int heightOfBinaryTreeInRecursion(BinaryTreeNode<T> root){ // 求二叉树的深度递归算法
        int leftHeight,rightHeight;
        if(root == null)
            return 0;
        leftHeight = heightOfBinaryTreeInRecursion(root.getLeft());
        rightHeight = heightOfBinaryTreeInRecursion(root.getRight());
        if(leftHeight > rightHeight)
            return (leftHeight+1);
        else 
            return (rightHeight+1);
    }

非递归算法

public static<T> int heightOfBinaryTree(BinaryTreeNode<T> root){ // 求二叉树深度的非递归算法
        if(root == null)
            return 0;
        DynArrayQueue<BinaryTreeNode<T>> queue = new DynArrayQueue<>();
        queue.enQueue(root);
        queue.enQueue(null); // 作为分层的标记
        int count = 0; // 第一层
        while(!queue.isEmpty()){
            root = queue.deQueue();
            if(root == null){ // 如果一层结束
                if(!queue.isEmpty()) // 为下一层添加标记
                    queue.enQueue(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/70046005
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞