剑指JZ86 在二叉树中找到两个节点的最近公共祖先

在二叉树中找到两个节点的最近公共祖先_牛客题霸_牛客网

描述

给定一棵二叉树(保证非空)以及这棵树上的两个节点对应的val值 o1 和 o2，请找到 o1 和 o2 的最近公共祖先节点。

数据范围：1 \le n \le 10001≤n≤1000，树上每个节点的val满足 0<val \le 1000<val≤100

要求：空间复杂度 O(1)O(1)，时间复杂度 O(n)O(n)

注：本题保证二叉树中每个节点的val值均不相同。

如当输入[3,5,1,6,2,0,8,#,#,7,4],5,1时，二叉树{3,5,1,6,2,0,8,#,#,7,4}如下图所示：

所以节点值为5和节点值为1的节点的最近公共祖先节点的节点值为3，所以对应的输出为3。

示例1

输入：

[3,5,1,6,2,0,8,#,#,7,4],5,1

返回值：3

一开始拿到题的时候我会想依次往上寻找，当两个节点找到同一个值了，它便是最近公共祖先，但是有个问题，先得遍历找到节点，并且把遍历经过的节点存储起来，当离节点远的时候，很难判断这个节点是否应该存储起来。

解题思路参考了二叉搜索树的最近公共祖先，因为二叉搜索树一定是左<根<右，从根节点遍历，有三种情况，一种情况是当根节点大于两个给出的节点，则这两个节点是在根节点的左子树，把左结点作为根节点继续往下查找，第二种情况是当根节点小于两个给出的节点，则这两个节点是在根节点的右子树，把右结点作为根节点继续往下查找，第三种是根节点处于中间，那么此时根节点就是最近公共祖先

同样的，本题也从根节点出发往下判断，假如当前节点为空就不用继续往下判断了，或者当前节点值等于其中某个节点值，那么当前节点就是最近公共祖先，如果都不满足，那么分别从左右继续往下遍历，当前节点有可能是最近公共祖先，也可能不是。如果是的话，往左能找到一个给定的节点值，往右也能找到一个给定的节点值，如果不是的话，两个节点要么同在左子树，要么同在右子树，返回对左右子树递归得到的非空的节点值

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @param o1 int整型 
     * @param o2 int整型 
     * @return int整型
     */
    int lowestCommonAncestor(TreeNode* root, int o1, int o2) {
        // write code here
        if(root==nullptr)
        {
            return 0;
        }
        if(root->val == o1 || root->val == o2)
        {
            return root->val;
        }
        int left = lowestCommonAncestor(root->left,o1,o2);
        int right = lowestCommonAncestor(root->right,o1,o2);
        if(left && right)
        {
            return root->val;
        }
        return left?left:right;
    }
};