如下一颗二叉树:
最远距离为红色路径,如何求得一颗二叉树的最远距离?
上面这棵二叉树,我们可以用子问题的方法解决,找到左子树的最远距离,再找到右子树的最远距离,相加即可
但是有特殊情况,比如下面这棵树:
再来看这棵树,如果按照上面的思想,左子树的最远距离为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);
}
我们在求最远距离的同时,把高度也就求出来了,就省去了第一个方法的调用求高度函数的步骤,节省了开销,但是核心思想是一样的!