记一次失败的面试手撕——求二叉树里任意两节点的距离

本文探讨了一次面试中遇到的问题,即如何在二叉树中计算任意两个节点之间的距离。通过理解二叉树的结构,实现一个有效的算法来解决这一挑战。
摘要由CSDN通过智能技术生成

实现函数,获取二叉树里面任意两个节点的距离:


```cpp
struct TreeNode{
	TreeNode* left;
	TreeNode* right;
};
int distance(TreeNode* proot,TreeNode* p,TreeNode* q)

```cpp
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
};
int distance(TreeNode* proot,TreeNode* p,TreeNode* q)
{
	if (proot == nullptr)
		return 0;

	TreeNode* father = FindFather(proot, p, q);
	int dis1 = rootDisNode(proot, p);
	int dis2 = rootDisNode(proot, q);
	int tmp = rootDisNode(proot, father);
	return dis1 + dis2 - 2 * tmp;
}
TreeNode* FindFather(TreeNode* proot, TreeNode* p, TreeNode* q)
{
	if (proot == nullptr)
		return nullptr;
	if (p == proot || q == proot)
		return proot;

	TreeNode* left = FindFather(proot->left, p, q);
	TreeNode* right = FindFather(proot->right, p, q);
	if (left && right)
		return proot;
	
	if (left != nullptr)
		return left;
	else
		return right;
}
int rootDisNode(TreeNode* proot, TreeNode* pNode)
{
	if (proot == nullptr)
		return -1;//找不到
	if (proot == pNode)
		return 0;

	int level = rootDisNode(proot->left, pNode);
	if (level == -1)//如果左边没找到,就找右边
	{
		rootDisNode(proot->right, pNode);
	}
	else//左边找到了
	{
		return level + 1;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值