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

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

思路一🧐：

  我们观察发现，祖先结点只会有两种情况：

1. p、q有一个是公共祖先结点，那么root自己本身就是祖先。
2. p、q总是在公共祖先的的左右两边。

  那么我们可以根据情况判断，需要用到递归，root为当前所开始的结点，p、q为需要寻找的结点。当p、q有一个等于root时，可以直接返回root。当p、q分布在两边时，我们可以用一个函数判断，去它的左子树和右子树寻找，找到了就返回true，没找到就返回false。

  而需要函数判断的情况有：p在左子树，p在右子树，q在左子树，q在右子树，我们需要将四种情况全部在函数中判断，最后出现p在左，q在右或者p在右q在左时，返回root结点，该结点为最近公共祖先。如果p、q在同一边，那么需要去同边再次寻找。

  不过这种方式时间复杂度会很高，因为有树全偏向一边的情况。

代码一🔎：

class Solution {
public:
    bool FindNode(TreeNode* root, TreeNode* x)
    {
        if(root == nullptr)
            return false;
        if(root == x)
            return true;
        return FindNode(root->left, x) || FindNode(root->right, x); //有一边找到就成功
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(p == root || q == root)
            return root;
        bool pInLeft,pInRight;
        bool qInLeft,qInRight;

        pInLeft = FindNode(root->left, p); //从该结点的左树找
        pInRight = !pInLeft; //左树没找到就一定在右树
        
        qInLeft = FindNode(root->left, q); //同上
        qInRight = !qInLeft;

        //q在右或者p在右q在左时，返回root结点
        if((pInLeft && qInRight) || (pInRight && qInLeft))
            return root;
        //都在左边时，再去左树找
        if(pInLeft && qInLeft)
            return lowestCommonAncestor(root->left, p, q);
        else
            return lowestCommonAncestor(root->right, p, q);
    }
};

思路二🧐：

  还是要找p、q，但是这次我们把找p、q的路径分别用栈保存下来，然后将p、q的size变为一致，此时p、q到公共祖先的步数相同，那么只需要一直pop，直到p、q相同，此时相同的结点就是最近公共祖先。

代码二🔎：

class Solution {
public:
    bool FindPath(TreeNode* root, TreeNode* x, stack<TreeNode*>& path)
    {
        if(root == nullptr)
            return false;
        
        path.push(root); //将结点push进去

        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); //获取p的路径
        FindPath(root, q, qPath); //获取q的路径

        while(pPath.size() != qPath.size()) //让size相同
        {
            if(pPath.size() > qPath.size()) //谁的size更大pop谁
                pPath.pop();
            else 
                qPath.pop();
        }

        while(pPath.top() != qPath.top()) //比较p、q
        {
            pPath.pop();
            qPath.pop();
        }
        return pPath.top();
    }
};