C++ 代码

/**
 * 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) {
       //1.有右孩子，则在右孩子的最左边的节点
       if(p->right){
           auto temp = p->right;
           while(temp->left){
               temp = temp->left;
           }
           return temp;
       }
       //2.没有右孩子，则看自己是否是父亲节点的左孩子，不是的话，父亲节点和自己往上迭代，一直迭代结束后还不是的话，就
       //没有后继节点
        auto temp = p->father;
        while(temp){
            if(temp->left == p)
                return temp;

            p = temp;
            temp = p->father;
        }
        return nullptr;
    }
};