代码随想录算法训练营第二十一天|LeetCode530 二叉搜索树的最小绝对差、LeetCode501 二叉搜索树中的众数

R.S.G.久夏2024-05-28 21:42

题1：

指路：LeetCode530 二叉搜索树的最小绝对差

思路与代码：

递归方面我们定义两个指针，一个当前节点指针cur，一个为当前节点的前一个节点pre。中序遍历二叉搜索树，如果当前节点为空则返回。不为空则返回当前节点和上一个节点的差，取较小值。代码如下：

cpp 复制代码

class Solution {
private:
    int result = INT_MAX;
    TreeNode* pre = NULL;
    void traversal(TreeNode* cur) {
        if  (cur == NULL) return ;
        traversal(cur->left);  // 左
        if (pre != NULL) {  // 中
            result = min (result, cur->val - pre->val);
        }    
        pre = cur;
        traversal(cur->right);
    }
public:
    int getMinimumDifference(TreeNode* root) {
    traversal(root);
    return result;
    }
};

题2：

指路：LeetCode501 二叉搜索树中的众数

思路与代码：

cpp 复制代码

class Solution {
private:
    int maxCount = 0; // 最大频率
    int count = 0; // 统计频率
    TreeNode* pre = NULL;
    vector<int> result;
    void searchBST(TreeNode* cur) {
        if (cur == NULL) return ;
        searchBST(cur->left);       // 左
                                    // 中
        if (pre == NULL) { // 第一个节点
            count = 1;
        } else if (pre->val == cur->val) { // 与前一个节点数值相同
            count++;
        } else { // 与前一个节点数值不同
            count = 1;
        }
        pre = cur; // 更新上一个节点
        if (count == maxCount) { // 如果和最大值相同，放进result中
            result.push_back(cur->val);
        }
        if (count > maxCount) { // 如果计数大于最大值频率
            maxCount = count;   // 更新最大频率
            result.clear();     // 很关键的一步，不要忘记清空result，之前result里的元素都失效了
            result.push_back(cur->val);
        }
        searchBST(cur->right);      // 右
        return ;
    }

public:
    vector<int> findMode(TreeNode* root) {
        count = 0;
        maxCount = 0;
        pre = NULL; // 记录前一个节点
        result.clear();
        searchBST(root);
        return result;
    }
};

今天没心情做了，到这就停一下吧。我得放松一下。