直接上题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
下图是该数组转换成二叉树的形状
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree
数组转二叉树方法Java代码,递归方法实现
public static TreeNode ArrToTree(Integer []arr, int index)
{
//申请一个根结点
TreeNode root = null;
if (index<arr.length) {
Integer value = arr[index];
if (value == null) {
return null;
}
//自顶向下构建二叉树的每一层
root = new TreeNode(value);
root.left = ArrToTree(arr, 2*index+1);
root.right = ArrToTree(arr, 2*index+2);
return root;
}
return root;
}
完成代码如下
package LeetCode236;
import java.util.*;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public class Solution {
static Map<Integer, TreeNode> parent = new HashMap<Integer, TreeNode>();
static Set<Integer> visited = new HashSet<Integer>();
public static void dfs(TreeNode root) {
if (root.left != null) {
parent.put(root.left.val, root);
dfs(root.left);
}
if (root.right != null) {
parent.put(root.right.val, root);
dfs(root.right);
}
}
public static TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
dfs(root);
while (p != null) {
visited.add(p.val);
p = parent.get(p.val);
}
while (q != null) {
if (visited.contains(q.val)) {
return q;
}
q = parent.get(q.val);
}
return null;
}
public static TreeNode ArrToTree(Integer []arr, int index)
{
TreeNode root = null;
if (index<arr.length) {
Integer value = arr[index];
if (value == null) {
return null;
}
root = new TreeNode(value);
root.left = ArrToTree(arr, 2*index+1);
root.right = ArrToTree(arr, 2*index+2);
return root;
}
return root;
}
public static void main(String[] args) {
Integer nums[]={3,5,1,6,2,0,8,null,null,7,4};
TreeNode root=ArrToTree(nums, 0);
TreeNode p=new TreeNode(5);
TreeNode q=new TreeNode(4);
System.out.println(lowestCommonAncestor(root,p,q).val);
}
}