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

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

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

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

思路：二叉搜索树，按照中序遍历则为有序数组，此题则转换为在一个有序数组上求最小值

class Solution {
private:
    vector<int> vec;
    void traversal(TreeNode* root) {
        if(root == NULL) return;
        traversal(root->left);  //左
        vec.push_back(root->val);
        traversal(root->right);
    }
public:
    int getMinimumDifference(TreeNode* root) {
        int result = INT_MAX;
        vec.clear();
        traversal(root);
        for(int i = 1; i < vec.size(); i++) {
            result = min(abs(vec[i] - vec[i - 1]), result);
        }
        return result;
    }
};

用两个指针替代创建数组

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

注：遇到在二叉搜索树上求最值、差值之类的，先思考一下二叉搜索树是有序的，尽量要利用好这一特点。

LeetCode 501 二叉搜索树中的众数

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

思路：同样利用中序遍历的有序特点，但是怎么做到仅遍历一遍二叉搜索树就得到结果数组呢？

class Solution {
private:
    int maxCount = 0;
    int count = 0;
    TreeNode* pre = NULL;
    vector<int> result;
    void searhBST(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.push_back(cur->val);
        }
        searchBST(cur->right);  //右
        return;
    }
public:
    vector<int> findMode(TreeNode* root) {
        count = 0;
        maxCount = 0;
        TreeNode* pre = NULL;
        result.clear();
        searchBST(root);
        return result;
    }
};

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

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

思路：回溯

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  { //  (left == NULL && right == NULL)
            return NULL;
        }

    }
};