二叉树任意两个结点最长距离

最新推荐文章于 2023-01-02 13:49:37 发布

一纸一清风

最新推荐文章于 2023-01-02 13:49:37 发布

阅读量347

点赞数

CC 4.0 BY-SA版权

分类专栏：牛客网刷题

本文链接：https://blog.csdn.net/Luckyboom_/article/details/101280475

牛客网刷题专栏收录该内容

5 篇文章

订阅专栏

这篇博客探讨了如何在给定的二叉树中找到两个节点的最大距离。文章指出，最大距离可能出现在根节点的同一侧，或者分布在两侧。通过递归计算左子树和右子树的最远节点距离以及高度，可以找到整个二叉树中两个节点的最大距离。并提供了解决问题的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一棵二叉树，求出树中两个结点距离的最大值。

首先明确距离最大的两个结点出现位置：1，同时在根结点的左子树中；2，同时在根结点的右子树中；3，左右子树中各有一个结点。

假设我们分别得到了根结点左子树中两个结点最长距离，右子树中两个结点最长距离（这是求整棵树中两个结点最长距离的子问题罢了，既然程序能够求出整棵树两个结点最长距离，那么求左右子树中的最长距离只不过是两次递归调用而已），还要得到根结点左子树高度（最低层为0），这个高度代表左子树中最底层结点到根结点距离（图四），同样可以得到右子树高度。

将左子树两个结点最长距离，有结点两个结点最长距离，左、右子树高度之和比较，取得三者中的最大值，即为这棵二叉树中两个结点距离的最大值。

代码如下：

typedef struct Node {
    struct Node *lchild;
    struct Node *rchild;
 
    //左右子树高度（最底层结点到当前结点距离）及左右子树最大高度
    int lHeight, rHeight, height;
}
 
int findMaxLen(Node *root) {
    int lMaxLen, rMaxLen;
 
    if(root == NULL) return 0;
 
    lMaxLen = findMaxLen(root->lchild);
    rMaxLen = findMaxLen(root->rchild);
 
    if(root->lchild == NULL) root->lHeight = 0;
    else root->lHeight = (root->lchild->height) + 1;
 
    if(root->rchild == NULL) root->rHeight = 0;
    else root->rHeight = (root->rchild->height) + 1;
 
    //以当前结点为根结点的树的最大高度（长度）
    root->height = root->lHeight > root->rHeight ? root->lHeight : root->rHeight;
 
    return max(lMaxLen, rMaxLen, root->lHeight + root->rHeight); //max求三者最大值
}

class LongestDistance {
public:
    int findLongest(TreeNode* root) {
        assert(root != NULL);
        int res = 0;
        int mt[501];
        find_longest_detail(root, mt, res);
        return res;
    }
private:
    void find_longest_detail(TreeNode* t, int *mt, int& res){
        int lmax = 0, rmax = 0;
        if(t->left != NULL){
            find_longest_detail(t->left, mt, res);
            lmax = mt[t->left->val];
        }
        if(t->right != NULL){
            find_longest_detail(t->right, mt, res);
            rmax = mt[t->right->val];
        }
        mt[t->val] = std::max(lmax, rmax) + 1;
        res = lmax + rmax + 1 > res ? lmax + rmax + 1 : res;
    }
};