剑指offer-39-二叉树的深度

最新推荐文章于 2024-07-21 08:45:36 发布

原创最新推荐文章于 2024-07-21 08:45:36 发布 · 195 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

算法

本文探讨了二叉树的深度计算与平衡二叉树的判断方法，介绍了递归方式计算树深度，并提出两种判断平衡性的策略：重复遍历与后序遍历一次性判断。

题目

输入一棵二叉树，求该树的深度。
从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，
最长路径的长度为树的深度。

二叉树的深度是从根结点到叶子结点的最长路径。若这个树只有一个根节点，则深度为1。

接下来看一下解题思路：

思路一：

若这个树只有一个根节点，则深度为1。如果根结点只有左子树而没有右子树，那么树的深度应该是其左子树的深度加1；同样如果根结点只有右子树而没有左子树，那么树的深度应该是其右子树的深度加1.如果既有左子树又有右子树，那该树的深度就是其左、右子树深度较大值加1。
代码示例：

public int TreeDepth(Tree root) {
        if (root == null) {
            return 0;
        }
		//左子树的深度
        int left = TreeDepth(root.lchild);
        //右子树的深度
        int right = TreeDepth(root.rchild);

        return (left >= right) ? (left + 1) : (right + 1);
    }

总结

这里需要注意的是

这里采用的是递归的方式统计树的深度。

基于这种思想可以判断判断树是不是平衡二叉树

题目

输入一棵二叉树的根结点，判断该树是不是平衡二叉树；如果某二叉树中任意结点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

思路一：

需要重复遍历结点多次的解法

在遍历树的每个结点的时候，调用统计树的深度方法得到它的左右子树的深度。如果每个结点的左右子树的深度相差都不超过1，那么就是一棵平衡二叉树。
代码示例：

public Boolean isBalancedTreeOne(Tree root) {
        if (root == null) {
            return true;
        }
        //统计每个结点左子树的深度
        int left = TreeDepth(root.lchild);
        //统计每个结点右子树的深度
        int right = TreeDepth(root.rchild);
        //求绝对值
        if (Math.abs(left - right) > 1) {
            return false;
        }

        return isBalancedTreeOne(root.lchild) && isBalancedTreeOne(root.rchild);
}

总结

这里需要注意的是

这种方法一个结点会被重复遍历多次，时间效率不高。

思路二：

每个结点只遍历一次的解法

用后序遍历的方式遍历二叉树的每个结点，在遍历到一个结点之前我们就已经遍历了它的左右子树。只要在遍历每个结点的时候记录它的深度（某一结点的深度等于它到叶结点的路径的长度）。就可以一边遍历一边判断每个结点是不是平衡的。
代码示例：

	//定义一个静态内部类，用于保存树的深度
	static class Depth{
        int n;
    }
    
    public boolean isBalancedTreeTwo(Tree root) {
        if (root == null) {
            return true;
        }
        Depth depth = new Depth();
        return balanced(root, depth);
    }

    private boolean balanced(Tree root, Depth depth) {
        if (root == null) {
            depth.n = 0;
            return true;
        }
		//统计左子树的深度
        Depth left = new Depth();
        //统计右子树的深度
        Depth right = new Depth();
		//左子树右子树都是平衡的
        if (balanced(root.lchild, left)
            && balanced(root.rchild, right)) {
            //统计当前结点的深度
            if (Math.abs(left.n - right.n) <= 1) {
                depth.n = 1 + ((left.n >= right.n) ? left.n : right.n);
                return true;
            }
        }
        return false;
}