二叉树的最近公共祖先[LeetCode]

题目描述

参见Leetcode官网。

思路分析

假定为普通二叉树，树的结构参考如下：

需要分3中情况考虑：

• 第一种根节点与其中一个节点值一致，则公共祖先即为根节点，比如待查找的节点为3、2，则必然为3
• 第二种待查找的节点为2、7，即2个节点均在左子树中，则根节点必然在左子树中，具体求取思路为：

先判定2、7 不与根节点3一致，进入3的左子树查找，5也不与2、7一致，进入5的左子树6，也不一致，走到头，返回null。进入5的右节点2，2与其中一个一致，2有可能是公共祖先，但不能保证一定是，需要结合3的右子树查找结果，当前情况，右子树走到最后也找不到与2、7一致的，因此右子树返回null，那么就可以确定2个值都在3的左子树，因此2必然是公共祖先

• 第三种待查找的节点为1、8，分析原理与第二种一致。

参考代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL) {
            return NULL;
        }
        if (p == root || q == root) {
            return root;
        }
        
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (left && right) {
            return root;
        }
        if (left) {
            return left;
        }
        return right;
    }
};
复制代码

补充非递归思路

从根节点到p的路径存入数组中，从根节点到q的路径存入数组中，然后遍历2个数组，比如p为6，q为4，
则根节点到p的路径为3->5->6； 根节点到4的路径为3->5->2->4
遍历到第三个节点时，一个为6，一个为2，开始不一样，则公共祖先必为第二个节点2.