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

求二叉树中节点的最大距离 ...
如果我们把二叉树看成一个图，
父子节点之间的连线看成是双向的，
我们姑且定义 " 距离 " 为两节点之间边的个数。
写一个程序，
求一棵二叉树中相距最远的两个节点之间的距离。

 

思路就是递归考虑左右子树的情况，记录左右子树最大距离的和为最后的结果

 

#include<iostream> using namespace std; int nMaxLen = 0; struct NODE { struct NODE* pLeft; struct NODE* pRight; int nMaxLeft; int nMaxRight; char chValue; }; //寻找树中最长的两段距离 void FindMaxLen(NODE* pRoot) { //遍历到叶子结点 if(pRoot == NULL) { return ; } //左子树为空，该节点的左边最长距离为0 if(pRoot->pLeft == NULL) { pRoot->nMaxLeft = 0; } //右子树为空，右边最长距离为0 if(pRoot->pRight == NULL) { pRoot->nMaxRight = 0; } //左子树不为空，递归寻找左子树最长距离 if(pRoot->pLeft != NULL) { FindMaxLen(pRoot->pLeft); } //右子树不为空，递归 if(pRoot->pRight != NULL) { FindMaxLen(pRoot->pRight); } //计算左子树最长节点距离 if(pRoot->pLeft != NULL) { int nTempMax = 0; if(pRoot->pLeft->nMaxLeft > pRoot->pLeft->nMaxRight) { nTempMax = pRoot->pLeft->nMaxLeft; } else { nTempMax = pRoot->pLeft->nMaxRight; } pRoot->nMaxLeft = nTempMax + 1; } //计算右子树最长节点距离 if(pRoot->pRight != NULL) { int nTempMax = 0; if(pRoot->pRight->nMaxLeft > pRoot->pRight->nMaxRight) { nTempMax = pRoot->pRight->nMaxLeft; } else { nTempMax = pRoot->pRight->nMaxRight; } pRoot -> nMaxRight = nTempMax + 1; } if((pRoot->nMaxLeft + pRoot->nMaxRight) > nMaxLen) { nMaxLen = (pRoot->nMaxLeft) + (pRoot->nMaxRight); } } NODE* creatBT()//建立一个二叉树的函数 { char ch; NODE* T; scanf("%c",&ch); if(ch=='.') { T = NULL;// . 代表空子树； } else { T=(NODE *)malloc(sizeof(NODE));//分配一个node的空间 //T = new NODE; //if(!T)exit(0); T->chValue = ch;//给T赋值 T->pLeft = creatBT(); T->pRight = creatBT();//给左子树赋值 } return T; } int main() { NODE* root; printf("请输按先序序列输入一串字符,当子树为空时,用.来代替/n"); root = creatBT(); printf("建树成功,T指向二叉树的根!/n"); cout << root->chValue << endl; FindMaxLen(root); cout << nMaxLen << endl; return 0; }