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

题目：

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

问题解析：

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

链接：

剑指Offer（第2版）：P65

思路标签：

数据结构：二叉搜索树

指针实现。

解答：

1. C++

• 分析树类的问题，最好绘制出二叉树的样例结构图，然后进行分析。
• 注意题目中是要求寻找中序遍历的下一个节点。
• 如果节点存在右子树：那么它的下一个节点就是右子树中的最左节点；
• 如果节点没有右子树：
  
  • 1. 节点为其父节点的左子节点，那么其父节点就是它的下一个节点；
  • 1. 节点为其父节点的右子节点，那么需要沿其父指针一直向上遍历，一直找到某个节点是其父节点的左子节点为止，那么这个节点的父节点即是需要寻找的下一个节点。

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

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