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

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

题目链接：530.二叉搜索树的最小绝对差
文章讲解：代码随想录|530.二叉搜索树的最小绝对差

思路

二叉搜索树中序遍历的有序性

代码

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;
    }
};

501.二叉搜索树中的众数

题目链接：501.二叉搜索树中的众数
文章讲解：代码随想录|501.二叉搜索树中的众数

思路

也是利用二叉搜索树中序遍历的有序性，在每次层if判断时候需要注意：

        if (pre == NULL) { // 第一个节点
            count = 1;
        } else if (pre->val == cur->val) { // 与前一个节点数值相同
            count++;
        } else { // 与前一个节点数值不同
            count = 1;
        }

如果这里的else if 是 if，那么当pre==NULL时这行也会运行，pre->val会报错空指针

        if (count == maxCount) { // 如果和最大值相同，放进result中
            result.push_back(cur->val);
        }

        if (count > maxCount) { // 如果计数大于最大值频率
            maxCount = count;   // 更新最大频率
            result.clear();     // 很关键的一步，不要忘记清空result，之前result里的元素都失效了
            result.push_back(cur->val);
        }

如果两个if判断顺序对调，maxCount = count;后count == maxCount成立，这样结果就往vec里push了两次。这边可以写成else if 就无所谓顺序了。
能else if 就 else if ，不要只写个if偷懒！！

代码

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;
    }
};

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

题目链接：二叉树的最近公共祖先
文章讲解：代码随想录|236. 二叉树的最近公共祖先
视频讲解：236. 二叉树的最近公共祖先

思路

后序遍历整个二叉树，如果遇到p或q就往上返回，当一个节点接收到的两个子树的返回值都不为空时，就是最近公共祖先
在写代码的时候，要把root(cur)想成不同的节点(空节点，叶子结点，非叶子结点等)，用不同的if来分别处理
如果是空节点或叶子结点，则跟终止条件有关
如果不是空节点且有儿子节点，则写在层处理中

代码

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {     
        if(!root) return nullptr;
        if(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;
        else if(!left && right) return right;
        else if(left && !right) return left;
        else return nullptr;
    }
};

今日收获

能else if 就 else if ，不要只写个if偷懒！！