每日一题(74) - 求二叉树中节点的最大距离

最新推荐文章于 2021-10-05 16:47:20 发布

原创最新推荐文章于 2021-10-05 16:47:20 发布 · 982 阅读

1 ·

CC 4.0 BY-SA版权

面试题专栏收录该内容

108 篇文章

订阅专栏

本文介绍了一种求解二叉树中两节点间最大距离的算法，通过后序遍历的方法结合节点深度来计算距离，给出了具体实现代码。

题目来自编程之美

题目

举例（图是网上偷的，在此谢过）

上例中，最大距离为4

思路

后序遍历 + 根据结点的深度求解以该节点为根的距离

该节点的距离等于其两个子树的最大深度之和

代码

struct BTNode
{
	int m_nData;
	BTNode* m_pLeft;
	BTNode* m_pRight;
};

int MaxDistance(BTNode* pRoot,int& nMaxDist)
{
	if (!pRoot)
	{
		return 0;
	}
	int nLeftDept = MaxDistance(pRoot->m_pLeft,nMaxDist);
	int nRightDept = MaxDistance(pRoot->m_pRight,nMaxDist);
	nMaxDist = max(nLeftDept + nRightDept,nMaxDist);
	return max(nLeftDept,nRightDept) + 1;
}

文中总结递归的思路很好。