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

题目描述:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。

注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。

 

中序遍历 :  左根右 的遍历顺序 

    会得到一个有序数组

因此我们需要判断当前节点的各种情况

情况一:剑指offer: 二叉树的下一个节点

情况二:

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

情况三:剑指offer: 二叉树的下一个节点

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)
		{
			pNode = pNode->right;
			while (pNode->left != nullptr)
			{
				pNode = pNode->left;
			}
			return pNode;
		}
		//如果当前节点的父节点不为空
		while (pNode->next != nullptr)
		{
			if (pNode->next->left == pNode)
			{
				return pNode->next;
			}
			pNode = pNode->next;
		}
		
		return nullptr;
	}
};