寻找二叉树某个节点的父节点(递归和非递归两种方法)java代码

寻找二叉树某个节点的父节点(递归和非递归两种方法)java代码


寻找二叉树某个节点的父节点(递归和非递归两种方法)java代码

1.非递归式,利用栈来辅助

public  TreeNode dfs(TreeNode root, TreeNode p){
   if(root == null) return null;  //树空,返回null
   Stack<TreeNode> stack =new Stack();
   stack.push(root);  //根节点入栈
   TreeNode fa = null;
   while (!stack.empty()){
      fa = stack.pop();  //出栈
      if(fa.left==p || fa.right == p) break; //出栈节点是否为p的父节点,是 则 break返回fa  
      if(fa.left!=null) stack.push(fa.left); 不是 fa的左右孩子入栈,下一轮循环  
      if(fa.right!=null) stack.push(fa.right);
   }
   return fa;

}

时间复杂度 O(N) : 其中 N为二叉树节点数;最差情况下,需要递归遍历树的所有节点。
空间复杂度 Olog(N) : 其中 N为二叉树节点数;节点出栈后,判断节点的右子树,右子树全部出栈后,左子树才会入栈。

优点: 1.找到父节点后即返回,不会再做多余的搜索。2.空间复杂度比较低。3.代码容易理解。

2.递归式

public TreeNode getParent(TreeNode root,TreeNode p){
   if(root == null ||root.left == p || root.right == p) return root; //树为空,或者p是root的子节点,返回root  

    TreeNode left = getParent(root.left,p); //root的左孩子 是否为p的父节点
    if(left != null) return left;  //是的话,return left,无需找右子树
    TreeNode right = getParent(root.right,p); //root的右孩子 是否为p的父节点
    if(right != null) return right;  //是的话,return right
    return left; //左右子树都不包含p,返回 null; (return right 同样是null)

 }

时间复杂度 O(N) : 其中 N为二叉树节点数;最差情况下,需要递归遍历树的所有节点。
空间复杂度 Olog(N) : 其中 N为二叉树节点数;递归深度达到 log(N) ,系统使用 log(N) 大小的额外空间。

优点: 1.代码比较简洁。2.不使用额外的数据结构。

注: 1.两种方法都采用的是深度优先遍历(dfs)的思想,同样适用于树的遍历。2.如果树严重不平衡(例如每个节点只有左子树)空间复杂度会退化为O(N)。

  • 19
    点赞
  • 69
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值