Lowest Common Ancestor of a Binary Tree

最新推荐文章于 2024-10-16 01:06:12 发布

原创最新推荐文章于 2024-10-16 01:06:12 发布 · 527 阅读

0 ·

CC 4.0 BY-SA版权

IT面试专栏收录该内容

18 篇文章

订阅专栏

本文介绍了二叉树中查找最近公共祖先的两种算法解法，第一种方法对于深度过高的情况可能无法通过，第二种方法则能够解决所有情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：点击打开链接

解法一：对于深度过高的case过不了。

/**
 * 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==NULL)return NULL;
        p1=p;
        p2=q;
        dfs(root,1);
        while(num1!=num2){
            if(num1>num2)num1=num1>>1;
            else num2=num2>>1;
        }
        ansNum=num1;
        getNode(root,1);
        return ansNode;
    }
private:
    int num1,num2,ansNum;
    TreeNode *p1,*p2,*ansNode;
    void dfs(TreeNode *root,int num){
        if(root){
            if(root==p1)num1=num;
            if(root==p2)num2=num;
            dfs(root->left,2*num);
            dfs(root->right,2*num+1);
        }
    }
    void getNode(TreeNode *root,int num){
        if(root){
            if(num==ansNum)ansNode=root;
            getNode(root->left,2*num);
            getNode(root->right,2*num+1);
        }
    }
};

解法二：全部能过。

/**
 * 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) {
        ans=NULL;
        dfs(root,p,q);
        return ans;
    }
private:
    TreeNode *ans;
    TreeNode* dfs(TreeNode *root,TreeNode *p,TreeNode *q){
        if(root&&ans==NULL){
            TreeNode* p1=dfs(root->left,p,q);
            TreeNode* p2=dfs(root->right,p,q);
            bool x=false,y=false;
            if(root==p||p1==p||p2==p)x=true;
            if(root==q||p1==q||p2==q)y=true;
            if(x&&y){
                ans=root;
                return NULL;
            }
            if(x)return p;
            if(y)return q;
            return NULL;
        }
        return NULL;
    }
};