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

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

思路： 中序遍历——左中右
1、存在右子树，找到右子树的最左子节点，即为下一个节点；
2、不存在右子树：
1）为左子节点，则下一节点是父节点；
2）为右子节点，则沿着它的父节点向上遍历，直到找到第一个父节点的左子节点为该节点的位置，则该节点的父节点就是下一个节点；
3）为右子节点，且找不到满足 2）的父节点，则该节点为尾节点。

class Solution {
public:
	TreeLinkNode* GetNext(TreeLinkNode* pNode)
	{
		if (pNode == nullptr)
			return nullptr;
		if (pNode->right != nullptr) {  //存在右子树
			pNode = pNode->right;
			while (pNode->left)
				pNode = pNode->left;
			return pNode;
		}
		while (pNode->next!=nullptr)   //不存在右子树
		{
			/*if (pNode->next->left == pNode)   //为左子节点，且不存在右子树
				return pNode->next;
			else    //为右子节点，且不存在右子树
			{
				TreeLinkNode* node = pNode->next;
				while (node->next != nullptr && node->next->left != node)
					node = node->next;
				return node->next;
			}*/
            TreeLinkNode* node = pNode->next;
            if (node->left == pNode)  //该节点的父节点的左子节点==该节点
				return node;
            pNode = pNode->next;
		}
		return nullptr;  //为尾节点
	}
};