[LeetCode] 1676. Lowest Common Ancestor of a Binary Tree IV

Given the root of a binary tree and an array of TreeNode objects nodes, return the lowest common ancestor (LCA) of all the nodes in nodes. All the nodes will exist in the tree, and all values of the tree's nodes are unique.

Extending the definition of LCA on Wikipedia: "The lowest common ancestor of n nodes p1, p2, ..., pn in a binary tree T is the lowest node that has every pi as a descendant (where we allow a node to be a descendant of itself) for every valid i". A descendant of a node x is a node y that is on the path from node x to some leaf node.

Example 1:

Input: root = [3,5,1,6,2,0,8,null,null,7,4], nodes = [4,7]
Output: 2
Explanation: The lowest common ancestor of nodes 4 and 7 is node 2.

Example 2:

Input: root = [3,5,1,6,2,0,8,null,null,7,4], nodes = [1]
Output: 1
Explanation: The lowest common ancestor of a single node is the node itself.

Example 3:

Input: root = [3,5,1,6,2,0,8,null,null,7,4], nodes = [7,6,2,4]
Output: 5
Explanation: The lowest common ancestor of the nodes 7, 6, 2, and 4 is node 5.

Constraints:

• The number of nodes in the tree is in the range [1, 104].
• -109 <= Node.val <= 109
• All Node.val are unique.
• All nodes[i] will exist in the tree.
• All nodes[i] are distinct.

二叉树的最近公共祖先 IV。

给定一棵二叉树的根节点 root 和 TreeNode 类对象的数组（列表） nodes，返回 nodes 中所有节点的最近公共祖先（LCA）。数组（列表）中所有节点都存在于该二叉树中，且二叉树中所有节点的值都是互不相同的。

我们扩展二叉树的最近公共祖先节点在维基百科上的定义：“对于任意合理的 i 值， n 个节点 p1 、 p2、...、 pn 在二叉树 T 中的最近公共祖先节点是后代中包含所有节点 pi 的最深节点（我们允许一个节点是其自身的后代）”。一个节点 x 的后代节点是节点 x 到某一叶节点间的路径中的节点 y。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/lowest-common-ancestor-of-a-binary-tree-iv
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题意是给了一棵二叉树和一个 nodes 数组，nodes 中的元素都存在于二叉树中。请找出这些节点的最小公共祖先。

思路是后序遍历，详细参见代码注释。

时间O(n)

空间O(n)

Java实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    TreeNode res = null;

    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode[] nodes) {
        // 把nodes放入hashset，这样可以以O(1)时间快速判断是否存在
        Set<Integer> set = new HashSet<>();
        for (TreeNode node : nodes) {
            set.add(node.val);
        }
        helper(root, set);
        return res;
    }

    private int helper(TreeNode root, Set<Integer> set) {
        // 统计当前子树内有多少节点存在于hashset
        int count = 0;
        if (root == null) {
            return 0;
        }
        if (set.contains(root.val)) {
            count++;
        }
        int left = helper(root.left, set);
        int right = helper(root.right, set);
        count += left;
        count += right;
        // 如果当前子树上的所有结点在set中都找到了且res为空，则res就是当前这棵子树
        if (count == set.size() && res == null) {
            res = root;
        }
        // 后序遍历往父节点返回的是当前子树内找到的存在于hashset内的节点个数
        return count;
    }
}

 

LeetCode 题目总结