题意:给定一棵二叉树,假设父子节点之间的连线看成时双向的,定义两个节点之间的边数为“距离”,现在需要求一棵二叉树中最远两个节点的距离
显然两个节点肯定是叶子节点
根据是否经过根节点,分两种情况考虑1.如果不经过根节点,那么最远的距离要么在左子树中,要么在右子树中
2.如果经过根节点,那么肯定一个节点在左子树中,另一个节点在右子树中
情况1可以由递归实现
在考虑情况2时已求得了根节点的左右子树中最远的两个节点的距离,同时也求得了左子树的左子树中的最长距离和左子树的右子树中的最长距离、右子树的左子树中的最长距离和右子树的右子树中的最长距离,利用得到的这些信息可以求得根节点的左子树的最长距离和根节点的右子树的最长距离,然后就可以得到经过根节点的最远两个节点的距离
取情况1、2中的最大值就可以求得一棵二叉树中最远两个节点的距离
#include <iostream>
using namespace std;
typedef struct BiTNode
{
struct BiTNode *lchild; //左子树
struct BiTNode *rchild; //右子树
int maxLeft; //左子树中的最长距离
int maxRight; //右子树中的最长距离
char data; //数值
}*BiTree, BiTNode;
int MaxLen = 0;
void FindMaxLen(BiTree pRoot)
{
//遍历到叶子节点
if(NULL == pRoot)
{
return;
}
//如果左子树为空,则左子树中的最长距离为0
if(NULL == pRoot->lchild)
{
pRoot->maxLeft = 0;
}
//如果右子树为空,则右子树中的最长距离为0
if(NULL == pRoot->rchild)
{
pRoot->maxRight = 0;
}
//如果左子树不为空,则求出左子树中最远两个节点的距离
if(pRoot->lchild != NULL)
{
FindMaxLen(pRoot->lchild);
}
//如果右子树不为空,则求出右子树中最远两个节点的距离
if(pRoot->rchild != NULL)
{
FindMaxLen(pRoot->rchild);
}
/*到现在为止,我们已求得了根节点左右子树中最远两个节点的距离,现在需要求经过根节点的最远两个节点的距离*/
//求得左子树中的最长距离
if(pRoot->lchild != NULL)
{
int tempMax;
if(pRoot->lchild->maxLeft > pRoot->lchild->maxRight)
{
tempMax = pRoot->lchild->maxLeft;
}
else
{
tempMax = pRoot->lchild->maxRight;
}
pRoot->maxLeft = tempMax + 1;
}
//求得右子树中的最长距离
if(pRoot->rchild != NULL)
{
int tempMax;
if(pRoot->rchild->maxLeft > pRoot->rchild->maxRight)
{
tempMax = pRoot->rchild->maxLeft;
}
else
{
tempMax = pRoot->rchild->maxRight;
}
pRoot->maxRight = tempMax + 1;
}
//对最长距离进行更新
if(pRoot->maxRight + pRoot->maxLeft > MaxLen)
{
MaxLen = pRoot->maxRight + pRoot->maxLeft;
}
}
//创建二叉树
void Union(BiTNode *node, int parent, int lchild, int rchild)
{
if(lchild != -1) //不为空
node[parent].lchild = &node[lchild];
if(rchild != -1) //不为空
node[parent].rchild = &node[rchild];
}
int main()
{
BiTNode nodes[9];
for(int i = 0; i < 9; ++i)
{
nodes[i].lchild = NULL;
nodes[i].rchild = NULL;
nodes[i].data = 'A' + i;
}
Union(nodes, 0, 1, 2);
Union(nodes, 1, 3, 4);
Union(nodes, 2, 5, 6);
Union(nodes, 3, 7, -1);
Union(nodes, 5, -1, 8);
FindMaxLen(nodes);
cout << "TheMaxLenBetweenNodeInBinaryTree is " << MaxLen << endl;
return 0;
}