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

题目

输出二叉树中指定结点的下一个结点（即中序遍历的后继）。给定树的根结点指针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;
}
};