剑指Offer第五十七题：二叉树的下一个结点

题目描述

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

思路：需要种类比较多，具体如下：（示例如图）

1. 如果该节点的右子树存在，则下一个节点就是右子树的最左。比如节点 a
2. 如果该节点右子树不存在，而且是父节点的左子树， 则下一个节点就是父节点，比如节点d
3. 如果该节点右子树不存在，而且是父节点的右子树，则下一个节点需要通过祖父节点判断，如果祖父节点存在，而且父节点是祖父节点的左子树，那么祖父节点就是下一个节点；但是如果父节点仍是祖父节点的右子树，则继续判断祖父节点的父节点，如果不为空，那么继续判断是否为左子树，以此循环，因为只有左子树才是下一个

 

代码如下：

/*
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 == NULL)
            return pNode;
        if(pNode->right != NULL)//右子树不为空，直接找右子树最左
        {
            TreeLinkNode* mid = pNode->right;
            while(mid->left != NULL)
                mid = mid->left;
            return mid;
        }
        if(pNode->next!=NULL)//若右子树为空，父节点不为空
        {
            TreeLinkNode* par = pNode->next;
            if(par->left == pNode)//若父节点的左子树是输入节点，直接返回父节点
                return par;
            if(par->right == pNode)//若父节点的右子树是输入节点
            {
                //判断父节点是祖父节点的左子树，若是则为祖父节点，否则就是继续寻找
                if(par->next != NULL)
                {
                    if(par->next->left == par)
                        return par->next;
                    else
                        par = par->next;
                }               
            }
        }      
        return NULL;
    }
};