二茶树找最远距离

如下一颗二叉树：

最远距离为红色路径，如何求得一颗二叉树的最远距离？

上面这棵二叉树，我们可以用子问题的方法解决，找到左子树的最远距离，再找到右子树的最远距离，相加即可

但是有特殊情况，比如下面这棵树：

再来看这棵树，如果按照上面的思想，左子树的最远距离为3，右子树的最远距离为0，相加后得3，但其实，红色路径才是这棵树的最远距离。所以：

我们需要分情况，我们要求出，左子树的最远距离和右子树的最远距离和经过根结点的最远距离（就是图一中红色的路径，经过了根结点），在这三者中，选出一个最大的就可以了，这样就解决了所有的情况

我们不难发现，最远距离其实就是树的高度，这样，我们就可以这样写代码：

int MAX(int a,int b,int c)
{
    if(a >= b && a >= c){
        return a;
    }
    if(b >= a && b >= c){
        return b;
    }
    return c;
}
int GetHeight(TreeNode *pRoot) //求树的高度
{
    if(pRoot == NULL){  //如果根结点为空，则返回0
        return 0;
    }

    int LeftH = GetHeight(pRoot->pLeft); //递归调用，求出左子树的高度
    int RightH = GetHeight(pRoot->pRight);//递归调用，求出右子树的高度

    return (LeftH > RightH ? LeftH : RightH) + 1; //左右子树中最大的+1就是整棵树的高度
                                                  //不要忘记+1
}

int TheFastestPath(TreeNode* pRoot)
{
    if(pRoot == NULL){   //若根结点为空，则返回0值
        return 0;
    }
    
    int LeftHeight = GetHeight(pRoot->pLeft); //求出左子树的高度
    int RightHeight = GetHeight(pRoot->pRight); //求出右子树的高度
    int RootPath = LeftHeight + RightHeight;  //左子树高度+右子树高度=经过根结点的最远距离
    int LeftPath = TheFastestPath(pRoot->pLeft); //递归调用，求出左子树的最远距离
    int RightPath = TheFastestPath(pRoot->pRight); //递归调用，求出右子树的最远距离

    return MAX(LeftPath,RightPath,RootPath); //三者中找出最大的就是整棵树的最远距离
}

上面的方法就可以找出一棵树中的最远距离，但是我们发现，这个方法递归调用的次树很多，还要去调用求高度的函数，在性能方面很差，所以我们可以对它进行优化，但思路还是一样的

int MAX(int a,int b,int c)
{
    if(a>=b&&a>=c){
        return a;
    }
    if(b>=a&&b>=c){
        return b;
    }
    return c;
}

int Optimize_TheFastestPath(TreeNode *pRoot,int *pHeight)
{
    if(pRoot == NULL){
        *pHeight = 0;
        return 0;
    }

    int LeftHeight,RightHeight;
    int LeftPath = Optimize_TheFastestPath(pRoot->pLeft,&LeftHeight);
    int RightPath = Optimize_TheFastestPath(pRoot->pRight,&RightHeight);
    int RootPath = LeftHeight + RightHeight;
    *pHeight = (LeftHeight > RightHeight ? LeftHeight : RightHeight) + 1;

    return MAX(LeftPath,RightPath,RootPath);
}

我们在求最远距离的同时，把高度也就求出来了，就省去了第一个方法的调用求高度函数的步骤，节省了开销，但是核心思想是一样的！