_1676.java

package com.fishercoder.solutions;

import com.fishercoder.common.classes.TreeNode;

import java.util.HashSet;
import java.util.Set;

public class _1676 {
    public static class Solution1 {
        /**
         * Since there are conditions for this problem: all values in the tree and given nodes are unique, we could simply use a HashSet to track the number of nodes we've found so far during the traversal.
         */
        TreeNode lca;

        public TreeNode lowestCommonAncestor(TreeNode root, TreeNode[] nodes) {
            Set<Integer> target = new HashSet<>();
            for (TreeNode node : nodes) {
                target.add(node.val);
            }
            dfs(root, target);
            return lca;
        }

        private int dfs(TreeNode root, Set<Integer> target) {
            if (root == null) {
                return 0;
            }
            int leftCount = dfs(root.left, target);
            int rightCount = dfs(root.right, target);
            int foundCount = leftCount + rightCount;
            if (target.contains(root.val)) {
                foundCount++;
            }
            if (foundCount == target.size() && lca == null) {
                lca = root;
            }
            return foundCount;
        }
    }
}