首页 > 代码库 > 二叉树中节点的最大距离(树的最长路径)——递归解法
二叉树中节点的最大距离(树的最长路径)——递归解法
上一篇文章说的是该题的一种变形,并给出了非递归解法。
现在我给出原题的一种递归解法。将会看到,现比较上篇博文,今天给出的递归解法的代码实现是相当简洁的。
问题描述:
如果我们把二叉树看成一个图,父子节点之间的连线看成是双向的,我们姑且定义"距离"为两节点之间边的个数。
写一个程序,求一棵二叉树中相距最远的两个节点之间的距离。测试用的树:
n1
/ \
n2 n3
/ \
n4 n5
/ \ / \
n6 n7 n8 n9
/ /
n10 n11
算法:
上篇博文我们用到了树的深度depth。而在递归解决此题的思考中,我发现用树的高度要比用深度简便得多。
这是因为:对于一个叶节点,它子树(尽管没有)的高度可以认为是0,它自己的高度是1,很容易区分。若是用深度,则都是0,会带来一些繁琐的判断。
题目就是求一棵树中的最长路径
对于节点t,以它为根的树的最长路径path一定是下列三个数中的最大值:
①t的左子树的最长路径lpath
②t的右子树的最长路径rpath
③t的左子树的高度+t的右子树的高度
——结论1
代码实现:
为了简洁优美,我尽量简化了代码,可能牺牲了一点易读性、增加了一些操作(如强行拼出来的那一长串return语句。。)
值得注意的是,程序中代码的顺序不能改变,因为对t->floor赋值的前提是t的左右子树的高度已知,它们由前两行递归代码已经顺带求出。因此顺序不能更改!!节点:
//节点结构体 struct BinaryTreeNode { BinaryTreeNode* left = NULL; BinaryTreeNode* right = NULL; int floor = 1; };
关键代码:
//查找最大路径,返回路径长度 int FindMaxPath(BinaryTreeNode* t) { if (t) { int lpath = FindMaxPath(t->left);//左子树最大路径 int rpath = FindMaxPath(t->right);//右子树最大路径 //t做根的树的层数等于子树最大层数+1 t->floor = max2((t->left) ? t->left->floor : 0, (t->right) ? t->right->floor : 0) + 1; //结论1 return max3(lpath, rpath, ((t->left) ? t->left->floor : 0) + ((t->right) ? t->right->floor : 0) ); } return 0; }
二叉树中节点的最大距离(树的最长路径)——递归解法