求一颗树任意两个节点的距离

最新推荐文章于 2023-10-28 01:48:11 发布

了不空蜗牛

最新推荐文章于 2023-10-28 01:48:11 发布

阅读量2.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：算法题文章标签：任意节点

本文链接：https://blog.youkuaiyun.com/qomoman/article/details/37930549

算法题专栏收录该内容

13 篇文章

订阅专栏

本文通过C++代码详细解释如何在树形结构中找到两个给定节点的最近公共祖先。包括代码实现、测试用例及解析过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个题可以先画树来看看，要知道它们共同的父亲就好了，可是共同父亲怎么知道呢？那我们从它的左子树和右子树找这两个节点。

来看个图比较好理解点吧，

如果当前节点是两个节点的共同父亲，或者是图中那种描述呢

直接来看C++代码吧

typedef struct Node{
	Node* left;
	Node* right;
	int val;
	Node(int v):val(v),left(NULL),right(NULL){}
}Node,*pNode;


int NodeDistance(pNode root,pNode first,pNode second,int &dist)
{
	if(root==NULL)
		return 0;
	int va=0;
	if(root==first||root==second)
		va=1;
	int valeft=NodeDistance(root->left,first,second,dist);
	int varight=NodeDistance(root->right,first,second,dist);
	if(!dist)
	{		
		if(valeft*va||varight*va||varight*valeft)
		{
			dist=varight+valeft;
			return 0;
		}
		else
			if(va||varight||valeft)
				return va?va:varight+valeft+1;
	}
	return 0;
}

最后给个测试用例吧

void  main()
{
	pNode root=new Node(1);
	root->left=new Node(2);
	root->right=new Node(3);
	root->left->left=new Node(6);
	root->left->right=new Node(7);
	root->right->left=new Node(8);
	root->right->right=new Node(9);
	root->left->left->right=new Node(10);
	root->left->left->left=new Node(11);
	root->right->left->left=new Node(12);
	root->right->left->left->right=new Node(13);

	int d=0;
	NodeDistance(root,root->right->left->left->right,root->left->right,d);
	while(1);
}