12.30 二叉树中等题

236. 二叉树的最近公共祖先

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

示例 1:

12.30 二叉树中等题_第1张图片

输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出:3
解释:节点 5 和节点 1 的最近公共祖先是节点 3 。

示例 2:

12.30 二叉树中等题_第2张图片

输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出:5
解释:节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。

思路1:

要让depth尽可能大,则如果从底向上去遍历节点,那第一个满足是p、q公共祖先的节点即是答案

细节:后序遍历是从底向上遍历二叉树

采用递归方法进行后序遍历:

1.递归结构:

        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
         //再判断该节点是否为P、q公共祖先节点

终止递归的条件:root==nullptr

判断是否为P、Q的祖先节点:判断left和right是否为null并重新写一个isChild进行递归判断,

class Solution {
public:
    bool isChild(TreeNode* root,TreeNode* target)
    {
        if(root==target) return true;
        else if(root==NULL) return false;
        return isChild(root->left,target) || isChild(root->right,target);
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        //再判断该节点是否为P、q公共祖先节点
        if(left!=NULL) return left;
        else if(right!=NULL) return right;
        else if(isChild(root,p)&&isChild(root,q)) return root;
        return NULL;
    }
};

这段代码的时间复杂度是O(N^2),其中N是二叉树中节点的数量。原因是在lowestCommonAncestor函数中,对于每个节点都会调用isChild函数,而isChild函数的时间复杂度是O(N),因为它需要递归遍历整个树来查找目标节点。

下面这个方法的复杂度为o(N)。

思路2:

代码随想录 (programmercarl.com)

要让depth尽可能大,则如果从底向上去遍历节点,那第一个满足是p、q公共祖先的节点即是答案

细节:后序遍历是从底向上遍历二叉树

采用递归方法进行后序遍历:

递归结构:

        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
         //再判断该节点是否为P、q公共祖先节点

终止递归的条件:root==nullptr或==p或==q

        if (root == q || root == p || root == NULL) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
   //再判断该节点是否为P、q公共祖先节点

判断该节点是否为P、q公共祖先节点可以利用返回的left、right值:

left、right的取值有四种情况:

1.left right 都为nullptr

则该root不是p q的祖先节点

2.left 为nullptr而right不为nullptr

则left不会是 p q的祖先节点,right可能是

3.right 为nullptr而left不为nullptr

则right不会是 p q的祖先节点,left可能是

4.left right 都不为nullptr

则root一定为 p 、q的公共祖先节点。该root返回到上一层递归中时,与该root对应的兄弟节点一定会是nullptr,则继续返回该root,直到第一层。

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == q || root == p || root == NULL) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        //再判断该节点是否为P、q公共祖先节点
        if (left != NULL && right != NULL) return root;
        if (left == NULL) return right;
        return left;
    }
};

在最坏情况下,需要遍历整个二叉树,因此时间复杂度是O(N),其中N是二叉树中节点的数量。

你可能感兴趣的:(每日一题,c++,数据结构)