陕西网站建设电话站长工具视频

二叉搜索树中的众数

力扣题目链接

题目描述

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

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

假定 BST 满足如下定义：

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

解题思路

这道题其实可以直接中序遍历然后结合数组找出众数即可，我是利用前k个高频元素来做，麻烦一些，但是可扩展性高，同时复习一下unorder_map，sort等用法，过了几天有点忘记了。
详细思路可以看我之前的文章：算法-栈和队列篇05-前 K 个高频元素

题解

class Solution {
public:static bool cmp(pair<int, int> x, pair<int, int> y){return x.second > y.second;}vector<int> findMode(TreeNode* root) {vector<int> ans;if(!root){return ans;}unordered_map<int, int> um;// 存储节点值和出现次数stack<TreeNode*> st;TreeNode* cur = root;while(!st.empty() || cur != NULL){if(cur != NULL){st.push(cur);cur = cur->left;}else{cur = st.top();st.pop();um[cur->val]++;cur = cur->right;}}// 对map排序，排序规则为按照出现次数vector<pair<int, int>> arr(um.begin(), um.end());sort(arr.begin(), arr.end(), cmp);// 最大的先放入答案数组中ans.push_back(arr[0].first);for(int i = 1; i < arr.size(); i++){// 循环判断前几个的出现次数是否相等，相等就加上if(arr[i].second == arr[i - 1].second){ans.push_back(arr[i].first);}else {break;}}return ans;}
};