寻找二叉树某个节点的父节点(递归和非递归两种方法)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)。