【代码随想录算法训练营14期】- day 21 第六章 二叉树part07

今日内容

● 530.二叉搜索树的最小绝对差
● 501.二叉搜索树中的众数
● 236. 二叉树的最近公共祖先

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

给你一个二叉搜索树的根节点 root ，返回 树中任意两不同节点值之间的最小差值 。

差值是一个正数，其数值等于两值之差的绝对值。

解题思路

二叉搜索树：中序遍历为递增数组，递归中序遍历后对数组循环计算最小值。
传递地址使用&

代码

class Solution {
public:
    // 中序遍历
    void getInorder(TreeNode* root,vector<int>& ans){
        if(!root) return;
        getInorder(root->left,ans);
        ans.push_back(root->val);
        getInorder(root->right,ans);
    }
    int getMinimumDifference(TreeNode* root) {
        vector<int> ans;
        getInorder(root,ans);
        if(ans.size() < 2) 
            return 0;
        int res = INT_MAX;
        for(int i = 1; i < ans.size(); i++){
           if(ans[i] - ans[i-1] < res){
               res = ans[i] - ans[i-1];
           }
       }
        return res;
    }
};

501.二叉搜索树中的众数

给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。

如果树中有不止一个众数，可以按 任意顺序 返回。

假定 BST 满足如下定义：

结点左子树中所含节点的值 小于等于 当前节点的值
结点右子树中所含节点的值 大于等于 当前节点的值
左子树和右子树都是二叉搜索树

解题思路

二叉搜索树使用中序遍历，先遍历左子树，再进行操作，再遍历右子树；
这里要找出所有众数；故返回的答案是数组；对每个节点，要判断其是不是与中序前一个的值相同，如果想同那么计数加一，否则前一个为空或者不同计数重新计数为1；计数操作以后判断数量是否等于最大计数，如果等于那么把当前值放入结果；如果大于最大计数，则答案数组clear，更改最大计数和答案；遍历完成下个节点时，pre赋值为当前节点；

代码

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

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

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。

解题思路

这里要查找二叉树最近的公共祖先；理解为在树中查找p、q的值，再由pq根节点回溯到同一根节点即为答案。自底向上故使用后序遍历。
对每个节点判断其为空返回空；当其为p或者q代表已找到其中一个那么就需要返回；先对左子树递归查找pq位置，再对右子树查找pq位置；如果左右子树都不为空，那么说明root就是最近的公共祖先，返回root；如果左子树不为空，右子树为空那么说明在左子树中找到了公共祖先

代码

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