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

本期我们对该题进行优化，不知道题目的小伙伴建议先看看之前的

236. 二叉树的最近公共祖先_KLZUQ的博客-CSDN博客

我们要将时间复杂度优化为O(N)

class Solution {
public:
    bool FindPath(TreeNode* root, TreeNode* x,stack<TreeNode*>& path)
    {
        if(root==nullptr)
            return false;
        path.push(root);
        if(root==x)
            return true;
        if(FindPath(root->left,x,path))
            return true;
        if(FindPath(root->right,x,path))
            return true;
        path.pop();
        return false;
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        stack<TreeNode*> pPath,qPath;
        FindPath(root,p,pPath);
        FindPath(root,q,qPath);
        while(pPath.size() > qPath.size())
        {
            pPath.pop();
        }
        while(pPath.size() < qPath.size())
        {
            qPath.pop();
        }
        while(pPath.top() != qPath.top())
        {
            pPath.pop();
            qPath.pop();
        }
        return pPath.top();
    }
};

思路：

我们那这个来举例，假设我们要找6和4的公共祖先

我们先看6，我们找一个栈，我们按照前序的顺序依次遍历，3不是，入栈，然后去3的左边，也就是5，5不是，继续入栈，然后去5的左边，是6，此时我们找到了，然后return true，返回到了5，继续return true，到了3，继续retrue true 

此时返回完毕，同时我们有了一个栈，这个栈里存储的是一个路径

 

接着我们找4，同样要先走3，5，6，然后继续找6的左边和右边，但是都为空，我们返回false，此时6的左右以及6本身都不是，那么我们pop掉栈里的一个元素，也就是pop掉了6，此时5的左边已经走完了，接着就是5的右边，用同样的顺序，我们入了2和7，然后7的左右都不是，7本身也不是，所以我们pop掉7，然后是2的右边，也就是4，我们找到了，然后依次返回true即可

 

此时我们有两个栈，然后我们比较他们的大小，谁大就pop谁，直到他们一样大为止，这里我们要pop下面的栈一个元素，栈里就还剩3，5，2，然后我们对比两个栈的元素，一样就返回这个节点，不一样就pop，上面的栈pop一个6，下面的pop一个2，然后栈顶元素都是5，也就是公共祖先，我们返回即可，时间复杂度为O(N)