二叉树的下一个结点

题目描述

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

 

理解了中序遍历，然后推导各种情况即可。

 

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 ->right == NULL)
        {
            if(!pNode ->next)
                return NULL;
            if(pNode ->next ->left == pNode)
                return pNode ->next;
            else
            {
                TreeLinkNode *curNode = pNode ->next ->next;
                TreeLinkNode *lastNode = pNode ->next;
                while(curNode && curNode ->left != lastNode)
                {
                    lastNode = curNode;
                    curNode = curNode ->next;
                }
                if(curNode == NULL)
                    return NULL;
                else
                    return curNode;
            }
        }
        else
        {
            TreeLinkNode *rightNode = pNode ->right;
            if(rightNode ->left == NULL)
                return rightNode;
            else
            {
                TreeLinkNode *node = rightNode ->left;
                while(node ->left)
                {
                    node = node ->left;
                }
                return node;
            }
        }
        return NULL;
    }
};