(C++)LeetCode#236. Lowest Common Ancestor of a Binary Tree

  • 题目:给定一个棵二叉树的两个节点o1/o2,求两个节点的最近公共祖先(LCA)
  • 难度:Medium
  • 思路:
    • 方法1.记录路径:先遍历二叉树,分别得到从根节点到o1/o2的两个路径,
    • 方法2.不记录路径:通过先序遍历二叉树,先判断当前遍历的节点是否为o1/o2,如果是则直接返回当前节点,否则继续遍历左子树、右子树,如果左子树返回结果和右子树返回结果同时存在,说明当前节点为最近公共祖先;否则判断左子树返回结果是否为空,如果不为空则返回左子树遍历结果,为空就返回右子树遍历结果
  • 代码:
方法一
class Solution {
public:
    bool getAncestor(TreeNode* root, TreeNode* node, vector<TreeNode*> &vec) {
        if (root == NULL) {
            return false;
        }
        if (root == node) {
            vec.push_back(root);
            return true;
        }
        if (getAncestor(root->left, node, vec) || getAncestor(root->right, node, vec)) {
            vec.push_back(root);
            return true;
        }
        return false;
    }

public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        vector<TreeNode*> lvec, rvec;//vector第一个节点是p或者q
        getAncestor(root, p, lvec);
        getAncestor(root, q, rvec);
        TreeNode* ancestor;
        int m = lvec.size();
        int n = rvec.size();
        while(m >0 && n > 0 && lvec[m-1]==rvec[n-1]){
            m--;
            n--;
        }
        return lvec[m];
    }
};
方法二
class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL || root == p || root == q) {
            return root;
        }
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (left && right) {
            return root;
        }
        return left?left:right;
    }
};

猜你喜欢

转载自blog.csdn.net/u012559634/article/details/80222344