题解 | #在二叉树中找到两个节点的最近公共祖先#

/*
 * function TreeNode(x) {
 *   this.val = x;
 *   this.left = null;
 *   this.right = null;
 * }
 */
/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 *
 * @param root TreeNode类
 * @param o1 int整型
 * @param o2 int整型
 * @return int整型
 */
function lowestCommonAncestor(root, o1, o2) {
    // write code here
    const tree1 = [];
    const tree2 = [];
    let o1Node;
    let o2Node;
    const traversal = (node) => {
        if (node.val === o1) {
            o1Node = node;
        }
        if (node.val === o2) {
            o2Node = node;
        }
        if (o1Node && o2Node) {
            return;
        }
        if (node.left) {
            node.left.parent = node;
            traversal(node.left);
        }
        if (node.right) {
            traversal(node.right);
            node.right.parent = node;
        }
    };
    traversal(root);
    while (o1Node) {
        tree1.push(o1Node);
        o1Node = o1Node.parent;
    }
    while (o2Node) {
        tree2.push(o2Node);
        o2Node = o2Node.parent;
    }
    const set = new Set();
    for (let i = 0; i < Math.max(tree1.length, tree2.length); i++) {
        if (tree1[i]) {
            if (!set.has(tree1[i])) {
                set.add(tree1[i]);
            } else {
                return tree1[i].val;
            }
        }
        if (tree2[i]) {
            if (!set.has(tree2[i])) {
                set.add(tree2[i]);
            } else {
                return tree2[i].val;
            }
        }
    }
}
module.exports = {
    lowestCommonAncestor: lowestCommonAncestor,
};

解题思路：首先遍历二叉树，并且给子节点添加一个指向父节点的指针，直到两个值对应的节点都找到就结束。然后根据值对应的节点构造两个数组，之后根据最大长度循环将节点存储在set对象里面，并且判断是否存在重复节点，第一个重复节点便是它们最近的祖节点