力扣面试题04.06后继者

设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回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节点就是单个节点,你前面的判断就呵呵啦

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值