二叉树_题目29:236. 二叉树的最近公共祖先

60 阅读3分钟

题目29:236. 二叉树的最近公共祖先

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

同类题目

前置知识

题目描述

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

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

示例 1:

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

示例 2:

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

示例 3:

 输入:root = [1,2], p = 1, q = 2
 输出:1

提示:

  • 树中节点数目在范围 [2, 105] 内。
  • -109 <= Node.val <= 109
  • 所有 Node.val 互不相同
  • p != q
  • pq 均存在于给定的二叉树中。

思路分析

递归+子问题

当前节点是最近公共祖先的条件

  • pqroot 的子树中,且分别在左、右子树中;
  • p=rootqroot 的左或右子树中;
  • q=rootp root 的左或右子树中;

因为要根据左右子树反馈的查询情况进行判断,因此,是后续遍历。

那么,当前节点要做那些事?

  • 我们从左右子树中寻找pq 找到就返回pq ,找不到返回nullptr
  • 如果左子树返回不为nullptr,右子树为nullptr, 则说明从左子树找到了pq,直接返回左子树的返回值。
  • 如果右子树返回不为nullptr ,左子树为nullptr,则说明从右子树找到了pq,直接返回右子树的返回值。
  • 如果左右子树返回都不为nullptr,则该节点就是最近公共祖先,直接返回。

程序代码

方法一

 // https://leetcode.cn/problems/lowest-common-ancestor-of-a-binary-tree/
 // 236. 二叉树的最近公共祖先
 ​
 #include<iostream>
 #include<vector>
 using namespace std;
 ​
 struct TreeNode {
     int val;
     TreeNode *left;
     TreeNode *right;
     TreeNode() : val(0), left(nullptr), right(nullptr) {}
     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 };
 ​
 class Solution {
 public:
     TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
         // 如果找到q 或者 q 返回q 或者 p
         if(root == p || root == q || root == nullptr)
         {
             return root;
         }
 ​
         TreeNode *leftv = lowestCommonAncestor(root->left,p,q);
         TreeNode *rightv = lowestCommonAncestor(root->right,p,q);
 ​
         // 左右都不为空,说明找到了p,q,直接返回当前节点
         if(leftv != nullptr && rightv != nullptr)
         {
             return root;
         }
 ​
         // 从左边找到,则返回找到的节点
         if(leftv != nullptr && rightv == nullptr)
         {
             return leftv;
         }
         // 从右边找到,则返回找到的节点
         else if(leftv == nullptr && rightv != nullptr)
         {
             return rightv;
         }else
         {
             return nullptr;
         }
         return nullptr;
     }
 };

复杂度分析

时间复杂度O(N),一共有 N个节点,最坏情况下,每个节点都要遍历一次。

空间复杂度O(N) ,递归栈的开销占用 O(N)

更多相关知识:github.com/pingguo1987…