【Leetcode笔记】236.二叉树的最近公共祖先

文章目录

题目要求

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

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

ACM

本题适合从下往上遍历,所以使用后序遍历来递归。

cpp 复制代码
#include <iostream>
#include <vector>
using namespace std;
// #include <unordered_map>
// #include <algorithm>

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) 
    {
        if (root == p || root == q || root == NULL)
        {
            return root;
        }    

        //左
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        
        //右
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        
        //根
        if(left == NULL) return right;
        if(right == NULL) return left;
        return root;
    }
};

int main(void)
{
    TreeNode* root = new TreeNode(8);
    root->left = new TreeNode(10);
    root->right = new TreeNode(4);
    root->left->left = new TreeNode(1);
    root->left->right = new TreeNode(7);
    root->right->left = new TreeNode(15);
    root->right->right = new TreeNode(20);
    root->left->right->left = new TreeNode(6);
    root->left->right->right = new TreeNode(5);
    Solution solution;

    TreeNode* p = root->left->right->left;
    TreeNode* q = root->left->right->right;
    TreeNode* result = solution.lowestCommonAncestor(root, p, q);
    cout << "Lowest Common Ancestor: " << result->val << endl;
    return 0;   
}

测试代码中 p、q 的定义,不能简单地定义一个根节点,TreeNode* p = new TreeNode(6);

TreeNode* p = new TreeNode(5);

运行结果

相关推荐
VertexGeek18 分钟前
Rust学习(八):异常处理和宏编程:
学习·算法·rust
石小石Orz19 分钟前
Three.js + AI:AI 算法生成 3D 萤火虫飞舞效果~
javascript·人工智能·算法
jiao_mrswang1 小时前
leetcode-18-四数之和
算法·leetcode·职场和发展
qystca1 小时前
洛谷 B3637 最长上升子序列 C语言 记忆化搜索->‘正序‘dp
c语言·开发语言·算法
薯条不要番茄酱1 小时前
数据结构-8.Java. 七大排序算法(中篇)
java·开发语言·数据结构·后端·算法·排序算法·intellij-idea
今天吃饺子1 小时前
2024年SCI一区最新改进优化算法——四参数自适应生长优化器,MATLAB代码免费获取...
开发语言·算法·matlab
是阿建吖!1 小时前
【优选算法】二分查找
c++·算法
王燕龙(大卫)2 小时前
leetcode 数组中第k个最大元素
算法·leetcode
Komorebi.py2 小时前
【Linux】-学习笔记05
linux·笔记·学习
不去幼儿园2 小时前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习