剑指offer-二叉树的下一个节点

问题

题目：[剑指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* GetNext(TreeLinkNode* pNode)
    {
        if(!pNode) return NULL;
        TreeLinkNode* right = pNode->right;
        if( right ) return leftbottom( right );
        else return firstfather( pNode, pNode->next );        
    }
private:
    TreeLinkNode* leftbottom(TreeLinkNode* r){
        while(r->left){
            r = r->left;
        }
        return r;
    }
    TreeLinkNode* firstfather(TreeLinkNode* child, TreeLinkNode* f){
        while(f){
            if( child == f->right ){
                child = f;
                f = f->next;
            }else return f;
        }
        return NULL;
    }
};