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

题目描述

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

本题知识点： 树

解答

中序遍历是先遍历左子结点，然后遍历中间结点，最后遍历右子结点。
有几种情况：
G结点（本身为左节点且左右子节点都没有）：父节点。
D结点（左右子节点都有且左右子节点无子结点）：下一个结点就是右子节点。
H结点（本身为右节点且左右子节点都没有且非最右结点）：父节点的父节点。
E结点（本身为左子节点且有右子节点）：右子节点
A结点（左右子节点都有且右子节点有子节点）：右子节点的第一个中序遍历结点
但这也很难判断。
网上找了一下别人归结的：
https://blog.csdn.net/mkfka/article/details/107053724

1、当前结点有右子节点，且其右子节点
（1）有最左子节点，则下一个结点为最左子节点，如图中的结点A的下一个结点为结点E
（2）无最左子节点，则下一个结点为当前结点的右结点，如图中的结点D的下一个结点为结点H
2、当前结点没有右子节点，且当前结点
（1）为左子节点，则下一个结点为其父节点，如图结点G的下一个结点为结点D
（2）为右子节点，则下一个结点为其祖宗里面作为左节点的父节点。如图 结点H的下一个结点为结点B

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;
		}
		while (pNode->next!=NULL) { //当前结点无右子节点
			if (pNode->next->left==pNode) {  //当前结点为左子节点
				return pNode->next;  //下一个结点为其父节点
			}
			pNode=pNode->next;//右子节点，则下一个结点为其祖宗里面作为左节点的父节点
		}
		return NULL;
    }
};