输出二叉树中中序遍历中给定结点的下一个结点

题目

输出二叉树中指定结点的下一个结点（即中序遍历的后继）。给定树的根结点指针TreeNode* root和结点的值int p，请返回值为p的结点的后继结点的值，若不存在后继返回-1。

分析

中序遍历的非递归写法。

第一步：中序遍历的非递归

    BTNode* p = root;  
    stack<BTNode*> s;  
    while (!s.empty() || p)  
    {  
        //一直遍历到左子树最下边，边遍历边保存根节点到栈中  
        while (p)  
        {  
            s.push(p);  
            p = p->lchild;  
        }  
        //当p为空时，说明已经到达左子树最下边，这时需要出栈了  
        if (!s.empty())  
        {  
            p = s.top();  
            s.pop();  
            cout << setw(4) << p->data;  
            //进入右子树，开始新的一轮左子树遍历  
            p = p->rchild;  
        }  
    }  

第二步：边遍历边进行比较，完整代码如下

class Successor {
public:
    int findSucc(TreeNode* root, int p) {
        // write code here
        if(root == nullptr)
            return -1;
        stack<TreeNode*> s;
        TreeNode *cur = root;
        bool flag = false;
        while(cur || !s.empty()){
            while(cur){
                s.push(cur);
                cur = cur->left;
            }
            if(!s.empty()){
                TreeNode *q = s.top();
                s.pop();
                if(flag)
                    return q->val;
                else if(q->val == p)
                    flag = true;
                cur = q->right;
            }
        }
        return -1;
    }
};