二叉树的下一个结点

题目描述

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

思路：主要清楚中序遍历的过程，中序遍历的顺序是左子树，根，右子树，所以当前结点的下一个结点要么就在右子树里，是右子树的最左子结点，如果右子树为空，那么就在父结点。如果当前结点是父结点的左子树，那么下一个结点就是父结点，如果是右子树就接着往上找就可以了。

代码：

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