求两个节点的公共祖先问题,直接描述起来会清楚一点:假设一棵树有两个节点2号和5号闯了祸,需要找到最低公共负责人来负责。
我们从根节点3号开始问"3节点,2号和5号都是你属下吗?"
3号遇到这个问题就先问左边的部下6号"2号和5号在你手下吗?"
6号遇到这个问题,然后发现左部下就是2号,然后问右部下11号"2号和5号在你手下吗?"
11号遇到这个问题,检查了左部下什么都没有,然后发现右部下就是5号,就返回给上级6号说"5号是我手下"
这时候6号就会发现自己的左部下产出了一个2号,自己的右部下又产出了一个5号,那自己就是那个最低公共负责人,就向上级3号报告说:“最低公共负责人就是6号”
3号收到6号的反馈,就直接跟我们返回说"最低公共负责人就是6号"
这是求解LCA的整个过程,主要是使用深度优先DSF来做。
上述的过程也可以通过视频来理解:https://www.youtube.com/watch?v=py3R23aAPCA
递归代码(JAVA描述)
public TreeNode LCA(TreeNode root, TreeNode p, TreeNode q) {
if(root==null||root==p||root==q) return root;
TreeNode left=LCA(root.left,p,q);
TreeNode right=LCA(root.right,p,q);
if(left!=null&&right!=null) return root;
return left!=null?left:right;
}
其中TreeNode这个类的结构为:
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
代码说明
递归到某一个节点时,可能会有以下一些情况出现:
- root根节点就是两个树节点中的其中一个,此时根节点就是LCA,直接返回root即可。
在代码上表现在:
if(root==null||root==p||root==q) return root;
- 如果root节点的左子树不存在p q这两个节点,那p q的LCA只能存在于root的右子树,在它的右子树查找即可。代码上可以表现为:
TreeNode left=LCA(root.left,p,q);
TreeNode right=LCA(root.right,p,q);
if(left==null) return right;
if(right==null) return left;
- 如果root节点左子树和右子树各有一个p q,则root就是这两个树节点的LCA,直接返回root。代码表现为:
if(left!=null&&right!=null) return root;
时间和空间复杂度
时间复杂度:
基本上每个树节点遍历了一遍,时间复杂度为O(n),n是树节点的数量
空间复杂度:
树越深,就需要越多空间存储递归的路径,空间复杂度为O(h),h是树的深度