代码随想录算法训练营第18天 | 530.二叉搜索树的最小绝对差、501.二叉搜索树中的众数、236. 二叉树的最近公共祖先

代码随想录算法训练营第18天 | 530.二叉搜索树的最小绝对差、501.二叉搜索树中的众数、236. 二叉树的最近公共祖先

---

530.二叉搜索树的最小绝对差

题目链接：530.二叉搜索树的最小绝对差

解题思路

本题是二叉搜索树，二叉搜索树是有序的，遇到二叉搜索树上求最值，求差值等问题，可以将其想成在一个有序数组上求最值、求差值。将其转化为一个有序数组，遍历一遍数组，就可以统计最小差值。

代码实现

class Solution {
public:
    vector<int> vec;
    void traversal(TreeNode* root) {
        if (root == nullptr)
            return;
        traversal(root->left);
        vec.push_back(root->val);
        traversal(root->right);
    }

    int getMinimumDifference(TreeNode* root) {
        vec.clear();
        traversal(root);
        if (vec.size() < 2)
            return 0;
        int result = INT_MAX;
        for (int i = 1; i < vec.size(); i++) {
            result = min(result, vec[i] - vec[i - 1]);
        }
        return result;
    }
};

题目总结

将二叉搜索树转化为有序数组，遍历数组，按题目要求解题。

501.二叉搜索树中的众数

题目链接：501.二叉搜索树中的众数

解题思路

首先将其转化为有序数组，然后对相邻的两个元素作比较，输出频率最高的元素。

使用 pre 指针和 cur 指针，定义一个指针指向前一个节点，这样 cur 可以和 pre进行比较。

遍历一遍数组，找出最大频率，然后再重新遍历一遍数组，把出现频率等于最大频率的元素放入集合。

代码实现

class Solution {
public:
    int count;
    int maxCount;
    vector<int> result;
    TreeNode* pre;
    void searchBST(TreeNode* cur) {
        if (cur == nullptr)
            return;
        searchBST(cur->left);
        if (pre == nullptr) {
            count = 1;
        } else if (pre->val == cur->val) {
            count++;
        } else {
            count = 1;
        }
        pre = cur;
        if (count == maxCount) {
            result.push_back(cur->val);
        }
        if (count > maxCount) {
            maxCount = count;
            result.clear();
            result.push_back(cur->val);
        }

        searchBST(cur->right);
        return;
    }

    vector<int> findMode(TreeNode* root) {
        count = 0;
        maxCount = 0;
        pre = nullptr;
        result.clear();
        searchBST(root);
        return result;
    }
};

题目总结

对于普通二叉树中的众数，可以使用 map 统计元素出现的频率并排序，取前面高频的元素的集合。

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

题目链接：236.二叉树的最近公共祖先

解题思路

自下而上查找可以找到公共祖先，而回溯的过程就是自下而上，后序遍历符合回溯的过程。

若找到一个节点，发现左子树出现节点 p，右子树出现节点 q，或者左子树出现节点 q，右子树出现节点 p，那么该节点就是节点 p 和 q 的最近公共祖先。

代码实现

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == q || root == p || root == NULL)
            return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (left != NULL && right != NULL)
            return root;
        if (left == NULL && right != NULL)
            return right;
        else if (left != NULL && right == NULL)
            return left;
        else {
            return NULL;
        }
    }
};

题目总结

1.求最小公共祖先，需要从底向上遍历，那么二叉树，只能通过后序遍历（即：回溯）实现从底向上的遍历方式。

2.在回溯的过程中，必然要遍历整棵二叉树，即使已经找到结果了，依然要把其他节点遍历完，因为要使用递归函数的返回值（也就是代码中的left和right）做逻辑判断。

3.要理解如果返回值left为空，right不为空为什么要返回right，为什么可以用返回right传给上一层结果。