《剑指offer》二叉树的下一个结点

最新推荐文章于 2020-05-14 16:00:17 发布

原创最新推荐文章于 2020-05-14 16:00:17 发布 · 1.4k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer #牛客网

《剑指offer》题目解析同时被 2 个专栏收录

66 篇文章

订阅专栏

《剑指offer》专题

65 篇文章

订阅专栏

本文介绍了一种高效算法，用于确定给定二叉树中某个节点在中序遍历下的下一个节点。主要考虑了三种情况：存在右子树、为父节点左子节点、为父节点右子节点的情况。

题目链接：http://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e?rp=3&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

题目描述
给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

思路

我们通过画一个二叉树来观察特点，很明显要寻找满足题意特点的下一个结点满足以下条件。

1.如果一个结点有右子树，那么其下一个结点就是右子树中最左的结点

2.如果一个结点没有右子树，同时该结点是父结点的左子结点，那么其下一个结点就是父结点

3.如果一个结点既没有右子树，而且还是父结点的右子结点，那么我们就要沿着父结点一层层往上遍历，直到找到一个新结点是其父结点的左子结点的时候，那么这个新结点的父结点就是我们要求的结点。

using namespace std;
/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {

    }
};
*/
class Solution
{
	public:
		TreeLinkNode* GetNext(TreeLinkNode* pNode)
		{
			if(pNode==nullptr)
				return nullptr;
			if(pNode->right!=nullptr)
			{
				TreeLinkNode *pRight = pNode->right;
				while(pRight->left!=nullptr)
					pRight = pRight->left;
				return pRight;
			}
			else if(pNode->next!=nullptr)
			{
				TreeLinkNode *pCurNode = pNode;
				TreeLinkNode *pParent = pNode->next;
				while(pParent!=nullptr && pCurNode == pParent->right)
				{
					pCurNode = pParent;
					pParent = pParent->next;
				}
				return pParent;
			}
		}
};