本期我们对该题进行优化,不知道题目的小伙伴建议先看看之前的
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)