给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。
import java.util.Scanner;
public class Main {
private static Node solve(Node root, Node h1, Node h2) {
if (root == null) {
return null;
}
if (h1 == root || h2 == root) {
return root;
}
Node left = solve(root.left, h1, h2);
Node right = solve(root.right, h1, h2);
if (left != null && right != null) {
return root;
}
return left == null ? right : left;
}
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
while (in.hasNext()) {
int n = in.nextInt();
Node[] nodes = new Node[n + 1];
for (int i = 1; i <= n; ++i) {
nodes[i] = new Node(i);
}
Node root = nodes[in.nextInt()];
for (int i = 1; i <= n; ++i) {
int fa = in.nextInt();
nodes[fa].left = nodes[in.nextInt()];
nodes[fa].right = nodes[in.nextInt()];
}
System.out.println(solve(root, nodes[in.nextInt()], nodes[in.nextInt()]).val);
}
}
}
class Node {
Node left;
Node right;
int val;
public Node(int val) {
this.val = val;
}
}