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