LeetCode——二叉搜索树中的顺序后继
Q:给你一个二叉搜索树和其中的某一个结点,请你找出该结点在树中顺序后继的节点。
结点 p 的后继是值比 p.val 大的结点中键值最小的结点。
示例 1:
输入: tree = [2,1,3], node = 1
输出: 2
解析: 1 的中序后继结点是 2 。注意节点和返回值都是 Node 类型的。
示例 2:
输入: tree = [5,3,6,2,4,null,null,1], node = 6
输出: null
解析: 该结点没有中序后继,因此返回 null 。
A:
- 如果当前节点有右孩子,找到右孩子,再持续往左走直到节点左孩子为空,直接返回该节点。
- 如果没有的话,就需要用到非递归的中序遍历。维持一个栈,当栈中有节点时:
- 往左走直到节点的左孩子为空,并将每个访问过的节点压入栈中。
- 弹出栈中节点,判断当前的前继节点是否为 p,如果是则直接返回当前节点。如果不是,将当前节点赋值给前继节点。
- 往右走一步。
- 如果走到这一步,说明不存在顺序后继,返回空。
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
if (p.right != null) {
p = p.right;
while (p.left != null) p = p.left;
return p;
}
Stack<TreeNode> stack = new Stack<>();
int inorder = Integer.MIN_VALUE;
while (!stack.isEmpty() || root != null) {
while (root != null) {
stack.push(root);
root = root.left;
}
root = stack.pop();
if (inorder == p.val) return root;
inorder = root.val;
root = root.right;
}
return null;
}
类似查找前驱:
如果当前节点的左子树不为空,那么该点的前驱节点为该点左子树中最右的节点
如果当前节点的左子树为空,那么该点的前驱节点为从该点往上延伸,如果延伸到的点为其父亲节点的右孩子,那么这个父亲节点就是该点的前驱节点
public TreeNode inorderPrecursor(TreeNode root, TreeNode p) {
if (p.left!= null) {
p = p.left;
while (p.right!= null) p = p.right;
return p;
}
Stack<TreeNode> stack = new Stack<>();
int inorder = Integer.MIN_VALUE;
while (!stack.isEmpty() || root != null) {
while (root != null) {
stack.push(root);
root = root.left;
}
root = stack.pop();
if (root== p.val) return inorder;
inorder = root.val;
root = root.right;
}
return null;
}