二叉树后继

二叉树后继

如果有指向父亲的结点，则：

1.如果当前结点有右儿子，或者当前结点是根结点，则后继结点为右子树的最左叶节点；
2.否则，如果当前结点是父结点的左儿子，则后继结点就是父结点；(其实是第三种情况的一个特例，即自己是第0代祖先，返回第一代祖先)
3.否则，向上遍历，直到n-1代祖先是n代祖先的左儿子，则后继结点为n代祖先；或者遍历到根节点后未找到符合的n代结点，则该结点为中序遍历的最后结点，没有后继。

时间复杂度为树的高度O(lgN)。

    struct TreeNode  
    {  
        char key;  
        TreeNode *left;  
        TreeNode *right;  
        TreeNode *parent;  
    };  

    TreeNode *FindSuccessor(TreeNode *x)  
    {  
        if(x->right!=NULL)  
        {  
            TreeNode *q=x->right;  
            while(q->left)  
                q=q->left;  
            return q;  
        }  
        TreeNode *p=x->parent;  
        while(p&&x==p->right)  
        {  
            x=p;  
            p=p->parent;  
        }  
        return p;  
    }  

如果没有父指针，则通过栈来实现中序遍历：

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};

class Successor {
public:
    int findSucc(TreeNode *root,int p){
        stack<TreeNode *> st;
        bool isFound=false;
        TreeNode *cur=root;
        while(cur||!st.empty()){
            if(cur!=NULL){
                st.push(cur);
                cur=cur->left;
            } else {
                cur=st.top();st.pop();
                if(isFound==true)return cur->val;
                if(cur->val==p)isFound=true;
                cur=cur->right;
            }
        }
        return -1;
    }
};

也可以通过递归实现：

class Successor {
public:
    int findSucc(TreeNode* root, int p) {
        // write code here
        bool sign=0;
        return findSucc1(root,p,sign);
    }
    int findSucc1(TreeNode* root,int p,bool &sign)
    {
        if(root==NULL)
            return -1;
        int left=findSucc1(root->left,p,sign);
        if(left!=-1)
            return left;
        if(sign==true)
            return root->val;
        if(root->val==p)
            sign=true;
        return findSucc1(root->right,p,sign);
    }
};