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

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode *father;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL), father(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* inorderSuccessor(TreeNode* p) {
        //就是求树的后继，算法时间复杂度是o(h),h为树的高度
        //树的中序遍历的后继，有两种情况，一种是当前节点有右儿子，则右子树最左侧的节点就是该节点的后继
        //第二种情况是，当前节点没有右儿子，则沿着father往上找，找到第一个是其father左儿子的节点，则该节点father为后继；
        if(p->right)
        {
            p=p->right;
            while(p->left) p=p->left;
            return p;
        }
        while(p->father&&p==p->father->right) p=p->father;
        return p->father;
    }
};