寻找二叉树中序遍历的下一个节点

寻找二叉树中序遍历的下一个节点

题目

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

分析：

首先需要了解中序遍历的顺序：

• 中序遍历左子树
• 遍历根结点
• 中序遍历右子树
  

对图中的数中序遍历序列则为 ： D B A E C F

因此参考该图，考虑如何确定一个节点的中序遍历的后一个节点

结论如下：

• 如果该节点拥有右子树，则下一个节点即为对右子树的中序遍历第一个节点
• 如果该节点没有右子树
  • 如果该节点是父节点的左子树，则下一个节点应当为父节点
  • 如果该子树是父结点的右子树，则向上走(不断更迭为自己的父结点)，直至出现该节点为父结点的左子树情况(如果这样情况存在，如果不存在则说明该节点为最后一个节点，因此之后为NULL即可)

很容易发现，对于该节点没有右子树的两种情况，在写代码时可以一并考虑。

/*
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==NULL) return NULL;
		if(pNode->right!=NULL) //有右子树，则下一个节点即为对右子树的中序遍历第一个节点
		{
			pNode=pNode->right;
			while(pNode->left!=NULL)
			{
				pNode=pNode->left;
			}
			return pNode;
		}
		else //没有右子树
		{
			while(pNode->next!=NULL)  //向上迭代直至出现该节点为父节点的左节点的情况
			{
				if(pNode->next->left==pNode)
					return pNode->next;
				pNode=pNode->next;
				
			}
			return NULL;
		} 
    }
};