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

题目描述：

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

本题知识点：

树

解题思路：

二叉树的中序遍历顺序为左->中->右，可以分为两种情况讨论。

1. 当前结点不包含右子树，则向上寻找其父结点，如果其父结点是在左子树部分，则直接返回其父结点，如果在右子树部分，则继续向上寻找父结点，直到到达需要的结点。
2. 当前结点包含右子树，如果当前结点的右子树不包含左子树，则直接返回当前结点的右子树。如果当前结点的右子树包含左子树，则寻找当前结点右子树的最后一个左子树结点。

代码：

/*
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* func1(TreeLinkNode* pNode)
    {
        if(!pNode->next)
            return NULL;
        if(pNode == pNode->next->left)
            return pNode->next;
        if(pNode == pNode->next->right)
            return func1(pNode->next);
    }
    
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(!pNode)
            return NULL;
        if(!pNode->right)
            return func1(pNode);
        if(pNode->right)
        {
            if(!pNode->right->left)
                return pNode->right;
            pNode = pNode->right;
            while(pNode->left)
                pNode = pNode->left;
            return pNode;
        }
    }
};