代码随想录DAY21：二叉树

wertvoll

于 2023-02-13 19:29:56 发布

阅读量64

点赞数

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_45845722/article/details/129015241

版权

二叉树

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

题目描述

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

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

思路

中序遍历二叉搜索树，得到有序数组，对有序数组遍历求最小差值即可。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> vec;
    void traverse(TreeNode* node)
    {
        if (node == NULL)   return;
        traverse(node->left);
        vec.push_back(node->val);
        traverse(node->right);
    }

    int getMinimumDifference(TreeNode* root) {
        //中序遍历得到有序数组
        vec.clear();
        traverse(root);
        if (vec.size() < 2) return 0;
        int res = INT_MAX;
        for(int i = 1; i < vec.size(); i++)
        {
            res = min(res, vec[i] - vec[i - 1]);
        }
        return res;
    }
};

501. 二叉搜索树中的众数

题目描述

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

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

假定 BST 满足如下定义：

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

思路

1、利用map存放数字及出现次数，遍历二叉搜索树更新map；
2、map转换为vector根据出现次数排序，得到出现频率最高的一个元素；
3、数组for循环，查看是否有多个众数。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    //必须写 static
    bool static cmp(const pair<int, int>& a, const pair<int, int>& b)
    {
        return a.second > b.second;
    }

    void traverse(TreeNode* node, unordered_map<int, int>& map)
    {
        if (node == NULL)   return;
        map[node->val]++;
        traverse(node->left, map);
        traverse(node->right, map);
        return;
    }

    vector<int> findMode(TreeNode* root) {
        unordered_map<int, int> map;
        traverse(root, map);
        vector<pair<int, int>> vec(map.begin(), map.end());
        sort(vec.begin(), vec.end(), cmp); // 给频率排个序
        vector<int> res;
        res.push_back(vec[0].first);
        for (int i = 1; i < vec.size(); i++)
        {
            if (vec[i].second == vec[0].second)
            {
                res.push_back(vec[i].first);
            }
            else
            {
                break;
            }
        }
        return res;
    }
};

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

题目描述

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

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

思路

递归法

1、参数及返回值

2、终止条件
root == NULL || root == p || root == q
//表示到达空节点或找到了节点，此时return，向上找祖先节点即可。

3、单层的逻辑
左右子树分别返回满足条件的节点，（若不存在满足条件的节点，则返回NULL），逐层向上，传到根节点。
根节点对左右子树传递回的结果进行处理，两边都传回非空节点，则根节点为共同节点；
若只有一侧传回非空节点，则最近公共祖先为该侧传回的节点；
若两边都传回空，则说明没有满足条件的祖先节点。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL || 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;

        if (left == NULL && right != NULL)  return right;
        else if (left != NULL && right == NULL) return left;
        else{
            return NULL;
        }
        }
};