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

最新推荐文章于 2024-09-18 16:42:25 发布

Coding_Physical

最新推荐文章于 2024-09-18 16:42:25 发布

阅读量120

点赞数

分类专栏：算法笔记

本文链接：https://blog.csdn.net/Coding_Physical/article/details/115287413

版权

算法笔记专栏收录该内容

37 篇文章

订阅专栏

int postOrder(TreeNode* root, TreeNode* p, TreeNode* q, TreeNode *&ans){
        
        if (!root) {
            return 0;
        }
        // 查看子结点的梳计个数
        int lcnt = postOrder(root->left, p, q, ans);
        int rcnt = postOrder(root->right, p, q, ans);
        // 利用子结点返回的信息来进行处理
        // 如果左边有一个，右边有一个，那么当前root就是最低公共祖先
        if (lcnt == 1 && rcnt ==1) {
            ans = root;
        }else if(lcnt ==1 || rcnt ==1){
            // 如果左边找到一个，或者右边找到一个
            // 并且我等于其中某一个结点p|q
            // 那么当前root就是最低公共祖先
            if (root == p || root == q) {
                ans = root;
            }
        }
        // 返回值为以root为根的子树, 统计里面的p,q结点的个数。
        return lcnt + rcnt + ((root ==p || root ==q)? 1 : 0);
    }
    
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        
        TreeNode *ans = nullptr;
        
        postOrder(root, p, q, ans);
        
        return  ans;
    }

递归

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

        TreeNode *l = lowestCommonAncestor(root->left,p,q);
        TreeNode *r = lowestCommonAncestor(root->right,p,q);
        if(!l) return r;
        if(!r) return l;
        if(l && r) return root;
        return NULL;
    }
};

题目