代码随想录算法训练营第十五天| 530.二叉搜索树的最小绝对差、501.二叉搜索树中的众数、236. 二叉树的最近公共祖先
530.二叉搜索树的最小绝对差
题目
给你一棵所有节点为非负值的二叉搜索树,请你计算树中任意两节点的差的绝对值的最小值。
题目链接
解法
class Solution
{
private:
vector<int> vec;
void traveral(TreeNode *root)
{
if (root == NULL)
return;
traveral(root->left);
vec.push_back(root->val);
traveral(root->right);
}
public:
int getMinimumDifference(TreeNode *root)
{
vec.clear();
traveral(root);
if (vec.size() < 2)
return 0;
int result = INT_MAX;
for (int i = 1; i < vec.size(); i++)
{
result = min(result, vec[i] - vec[i - 1]);
}
return result;
}
};
501.二叉搜索树中的众数
题目
给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。
假定 BST 有如下定义:
结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
解法
class Solution
{
private:
void searchBST(TreeNode *cur, unordered_map<int, int> &map)
{
if (cur == NULL)
return;
map[cur->val]++;
searchBST(cur->left, map);
searchBST(cur->right, map);
return;
}
bool static cmp(const pair<int, int> &a, const pair<int, int> &b)
{
return a.second > b.second;
}
public:
vector<int> findMode(TreeNode *root)
{
unordered_map<int, int> map; // key:元素,value:出现频率
vector<int> result;
if (root == NULL)
return result;
searchBST(root, map);
vector<pair<int, int>> vec(map.begin(), map.end());
sort(vec.begin(), vec.end(), cmp); // 给频率排个序
result.push_back(vec[0].first);
for (int i = 1; i < vec.size(); i++)
{
// 取最高的放到result数组中
if (vec[i].second == vec[0].second)
result.push_back(vec[i].first);
else
break;
}
return result;
}
};
236. 二叉树的最近公共祖先
题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
解法
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;
}
}
};