剑指Offer-8：二叉树的下一个节点

题目：

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

问题解析：

寻找二叉树的中序遍历下一个节点，注意考虑不同的情况。

链接：

剑指Offer（第2版）：P65

思路标签：

数据结构：二叉搜索树

指针实现。

解答：

1. C++

分析树类的问题，最好绘制出二叉树的样例结构图，然后进行分析。

注意题目中是要求寻找中序遍历的下一个节点。

如果节点存在右子树：那么它的下一个节点就是右子树中的最左节点；

如果节点没有右子树：

节点为其父节点的左子节点，那么其父节点就是它的下一个节点；

节点为其父节点的右子节点，那么需要沿其父指针一直向上遍历，一直找到某个节点是其父节点的左子节点为止，那么这个节点的父节点即是需要寻找的下一个节点。

/*
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;

TreeLinkNode* pNext = nullptr;
if(pNode->right != nullptr){
TreeLinkNode* pRight = pNode->right;
while(pRight->left != nullptr)
pRight = pRight->left;

pNext = pRight;
}
else if(pNode->next != nullptr){
TreeLinkNode* pCurrent = pNode;
TreeLinkNode* pParent = pNode->next;
while(pParent != nullptr && pCurrent == pParent->right){
pCurrent = pParent;
pParent = pParent->next;
}

pNext = pParent;
}

return pNext;
}
};

细心，注意指针操作的正确性。