Interview100-11 求二叉树中节点的最大距离

最新推荐文章于 2020-03-17 23:10:32 发布

Ethan_pika

最新推荐文章于 2020-03-17 23:10:32 发布

阅读量252

点赞数

分类专栏：面试算法100题文章标签： algorithm interview

本文链接：https://blog.csdn.net/yitian_z/article/details/88091532

版权

面试算法100题专栏收录该内容

50 篇文章 0 订阅

订阅专栏

题目

如果我们将二叉树看成一个图，父子节点之间的连线看成双向的边，我们定义”距离“为两个节点之间边的个数。设计一个算法，求出一个二叉树中相距最远的两个节点之间的距离。

解法

思路：二叉树中两个结点之间的最大距离，是如下两种情况中的最大值，第一种为：根结点到叶子节点的深度最大值。第二种为两个叶子节点之间的距离最大值。因此计算出二叉树中所有节点间的距离，取最大值。

GET-MAX-DISTANCE(Node root):
    int distance = 0;
    if root != null:
        // 计算以当前节点为根结点的最大距离
        distance = GET-DEPTH(root.left) + GET-DEPTH(root.right);
        // 计算当前节点的左/右子树的最大距离
        int left_distance = GET-MAX-DISTANCE(root.left);
        int right_distance = GET-MAX-DISTANCE(root.right);

        distance = left_distance > distance ? left_distance : distance;
        distance = right_distance > distance ? right_distance : distance;
    return distance;

GET-DEPTH(Node root):
    int depth = 0;
    if root != null:
        int leftDepth = GET-DEPTH(root.left);
        int rightDepth = GET-DEPTH(root.right);
        depth = leftDepth > rightDepth ? leftDepth : rightDepth;
        depth++;
    return depth;

其中GET-DEPTH方法递归的计算以一个节点为根结点时的树的深度，它就等于左右子树深度的最大值+1。GET-MAX-DISTANCE方法为计算二叉树中节点最大距离的方法。它实际上以root节点为根结点时的最大距离，等于左右子树的深度和，然后与分别算出的左子树中的最大距离和右子树的最大距离，三者比较，就是当前二叉树中的最大距离。