剑指offer--------二叉树的下一个节点

最新推荐文章于 2022-04-18 15:24:45 发布

原创最新推荐文章于 2022-04-18 15:24:45 发布 · 207 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于确定给定二叉树节点在中序遍历中的下一个节点。根据节点的不同情况，如是否有右子树、是否为父节点的左子树等，提供了具体的实现思路与代码。

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

题目：给定一棵二叉树和其中的一个节点，找到中序遍历序列中该节点的下一个节点，树中节点有指向左孩子、右孩子及父节点的三个指针。

思路：根据当前节点在树中的位置，分为3种情况：

1、给定节点有右子树，则下一个节点为其右子树的最左节点；

2、给定节点没有右子树，但该节点是其父节点的左子树，则下一个节点为当前节点的父节点；

3、给定节点没有右子树且是其父节点的右子节点，则沿着指向父节点的指针一直向上遍历，直到找到一个是它父节点的左子节点的节点，若这样的节点存在，则这个节点的父节点就是给定节点的下一个节点。

实现：

struct BinaryTreeNode
{
	int value;
	BinaryTreeNode* parent;   //指向父节点
	BinaryTreeNode* lchild;
	BinaryTreeNode* rchild;
};
BinaryTreeNode* GetNext(BinaryTreeNode* pNode)
{
	BinaryTreeNode* pNext = NULL;
	if (!pNode) return NULL;
	if (pNode->rchild)
	{
		BinaryTreeNode* temp = pNode->rchild;
		while (temp)temp = temp->lchild;
		pNext = temp;
	}
	else if(pNode->parent)
	{
		BinaryTreeNode* curNode = pNode;
		BinaryTreeNode* parentNode = pNode->parent;
		//沿着父节点一直遍历，直到找到是父节点左子节点的节点为止
		while (parentNode && curNode == parentNode->rchild)
		{
			curNode = parentNode;
			parentNode = parentNode->parent;
		}
		pNext = parentNode;
	}
	return pNext;
}