二叉树的最近公共祖先 IV

/*
题目思路：
1. 与普通的Java解法相同，但在处理节点集合时使用流（Stream）API。
2. 将节点的值收集到一个Set中，然后递归寻找LCA。
*/

import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode[] nodes) {
        Set<Integer> values = Stream.of(nodes).map(node -> node.val).collect(Collectors.toSet());
        return findLCA(root, values);
    }
    
    private TreeNode findLCA(TreeNode root, Set<Integer> values) {
        if (root == null) return null;
        if (values.contains(root.val)) return root;
        TreeNode left = findLCA(root.left, values);
        TreeNode right = findLCA(root.right, values);
        if (left != null && right != null) return root;
        return left != null ? left : right;
    }
}