数据结构与算法（作业二十八）

最新推荐文章于 2024-04-24 22:20:45 发布

想扣篮的Phineas

最新推荐文章于 2024-04-24 22:20:45 发布

阅读量148

点赞数

本文链接：https://blog.csdn.net/BryceFei/article/details/105902714

版权

二叉树的最近公共祖先

在这里插入图片描述

递归，对每个节点对应的子树，若该子树不含有p或q，返回nullptr；否则，如果p和q分别位于当前子树根节点两侧，则返回当前节点，否则（p和q在同一侧，或者只有某一侧有p或q）返回来自左边或右边的LCA
运算代码如下：

class Solution {
public:
    //这种思路要求p、q一定要在树里，因为有可能找到一个节点p或q就返回了
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (!root || 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;
    }
};

在这里插入图片描述
递归，中序遍历，对每个节点，计算其左子树和右子树包含的p或q个数，并返回。第一个包含2个目标值的子树对应的根节点就是最低公共祖先。

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        int dummy = contain_counter(root, p, q);
        return ans;
    }
    int contain_counter(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (!root) return 0;
        int mid = 0;
        if (root == p || root == q) mid = 1;
        int left = contain_counter(root->left, p, q);
        if (mid + left == 2) {  //可能的话提前剪枝，避免无意义的遍历
            if (!ans) ans = root;
            return 2;
        }
        int right = contain_counter(root->right, p, q);
        if (left + right + mid == 2) if (!ans) ans = root;
        return left + right + mid;
    }
private:
    TreeNode* ans = nullptr;
};