题目描述
给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。
输入描述:
第一行输入两个整数 n 和 root,n 表示二叉树的总节点个数,root 表示二叉树的根节点。
以下 n 行每行三个整数 fa,lch,rch,表示 fa 的左儿子为 lch,右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子,rch同理)
最后一行为节点 o1 和 o2。
输出描述:
输出一个整数表示答案。
示例1
输入
8 1
1 2 3
2 4 5
4 0 0
5 0 0
3 6 7
6 0 0
7 8 0
8 0 0
4 5
输出
2
解法一:递归
import java.io.*;
import java.util.*;
public class Main{
public static void main(String[] args) throws Exception{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String[] info = br.readLine().trim().split(" ");
int len = Integer.parseInt(info[0]);
int r = Integer.parseInt(info[1]);
TreeNode[] nodes = new TreeNode[len+1];
for(int i=1;i<=len;i++){
nodes[i] = new TreeNode(i);
}
createTree(nodes,br);
String[] ss = br.readLine().trim().split(" ");
TreeNode node1 = nodes[Integer.parseInt(ss[0])];
TreeNode node2 = nodes[Integer.parseInt(ss[1])];
TreeNode root = nodes[r];
TreeNode res = findCommon(root,node1,node2);
if(res!=null){
System.out.println(res.val);
}
}
public static TreeNode findCommon(TreeNode root,TreeNode node1,TreeNode node2){
if(root==null||root==node1||root==node2) return root;
TreeNode left = findCommon(root.left,node1,node2);
TreeNode right = findCommon(root.right,node1,node2);
if(left!=null&&right!=null){
return root;
}
return left!=null?left:right;
}
public static void createTree(TreeNode[] nodes,BufferedReader br) throws Exception{
for(int i=0;i<nodes.length-1;i++){
String[] ss = br.readLine().trim().split(" ");
int data = Integer.parseInt(ss[0]);
int left = Integer.parseInt(ss[1]);
int right = Integer.parseInt(ss[2]);
if(left!=0){
nodes[data].left = nodes[left];
}
if(right!=0){
nodes[data].right = nodes[right];
}
}
}
}
class TreeNode{
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val){
this.val = val;
}
}