设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回null。
示例 1:
输入: root = [2,1,3], p = 1
2
/ \
1 3
输出: 2
示例 2:
输入: root = [5,3,6,2,4,null,null,1], p = 6
5
/ \
3 6
/ \
2 4
/
1
输出: null
吼吼,继续更新,终于找到了答题人数比较少的啦,
我来理下思路,待会客官自己研究下代码
首先说明整体思路是用的迭代
无非三种大情况:
1、其值比根节点值要大,那么也就意味着这个节点在根节点的右子树,我们由中序后继得知,这样的情况,返回值只会出现在右子树里面。这就可以把右子树给单独看做一个树我们调用相同的算法去操作;
2、其值与根节点相同,那么也就意味着其中序后继为根节点右子树的最左边
3、其值比根节点要小,那么这个的情况就有些复杂了
3.1其值和根节点的左孩子相同,且左孩子无右孩子,直接返回根节点
3.2其值和根节点的左孩子相同且左孩子有右孩子,那么我们需要找到左孩子的右孩子的最左节点
3.3若其值小于根节点左孩子,则继续迭代
3.4若其值大于根节点左孩子,则迭代的返回值若为空则则返回根节点,否则返回返回值。
第3种情况有点饶,大家自己画画树的图就好啦。
package medium;
public class succersor {
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
TreeNode re=null;
if(root.val<p.val){//在右子树
root=root.right;
if(root!=null){
re= inorderSuccessor(root, p);
}else
return null;
}
else if(root.val==p.val){
if(root.right!=null){
re=findleft(root.right);
}
return re;
}
else if(root.val>p.val){
if(root.left!=null&&root.left.right==null&&root.left.val==p.val){
return root;
}
else if(root.left!=null&&root.left.right!=null&&root.left.val==p.val){
return findleft(root.left.right);
}
else if(root.left!=null&&root.left.val<p.val){
re =inorderSuccessor(root.left, p)==null? root:inorderSuccessor(root.left, p);
}
else if(root.left!=null&&root.left.val>p.val){
re=inorderSuccessor(root.left, p);
}else
return null;
}
return re;
}
private TreeNode findleft(TreeNode root) {
while(root.left!=null){
root=root.left;
}
return root;
}
public static void main(String[] args) {
TreeNode root=new TreeNode(5);
TreeNode r=new TreeNode(6);
TreeNode r2=new TreeNode(2);
TreeNode r3=new TreeNode(4);
root.left=new TreeNode(3);
root.left.left=new TreeNode(1);
root.left.right=r3;
root.left.left.right=r2;
root.right=r;
succersor s=new succersor();
System.out.println(s.inorderSuccessor(root, new TreeNode(4)).val);
}
}
public TreeNode inorderSuccessor2(TreeNode root, TreeNode p) {
// p存在右子树,直接后继就是右子树的最左节点
if(p.right!=null){
p=p.right;
while(p.left!=null){
p=p.left;
}
return p;
}
//p不存在右子树
TreeNode node=root;
TreeNode res=null;
while(p!=node){
if(p.val<node.val){
// node比p大,表示node在p的后继路径上
res=nodel;
node=node.left;
}else{
// node比p小,表示node在p的前驱路径上
node=node.right;
}
}
return res;
}
OK,我从题解里看到了另一个解法,他是用p节点是否有孩子进行判断的。
但我认为这种做法不够好,若是p节点就是单个节点,你前面的判断就呵呵啦