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

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

        二叉树的特性是两个相邻节点之间可能存在最小绝对差，所以可以中序遍历同时计算节点间的差值，再分别比较得出答案。

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

501.二叉搜索树中的众数

        和上一题一个类型，可以为每个节点保存一个计数，计数最大的节点值放入数组中；要注意的是搜索到更大的计数时要将之前的数组清空。 

class Solution {
public:
    vector<int> res;
    TreeNode* pre; 
    int count = 0;
    int maxCount = INT_MIN;
    void traversal(TreeNode* root) {
        if(!root) return;
        if(root->left) traversal(root->left);
        if(pre == nullptr) {
            count = 1;
        }else if(root->val == pre->val) {
            count++;
        }else {
            count = 1;
        }
        pre = root;
        if(maxCount < count) {
            maxCount = count;
            res.clear();
            res.push_back(root->val);
        }else if(maxCount == count) {
            res.push_back(root->val);
        }
        if(root->right) traversal(root->right);
    }
    vector<int> findMode(TreeNode* root) {
        traversal(root);
        return res;
    }
};

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

        这题看了答案才有思路。自底向上（后序遍历），然后判断根节点和其公共祖先的位置关系。这里递归传递参数用的很妙。

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