二叉搜索树中的众数-树501-C++
程序员文章站
2022-02-28 06:19:58
...
算法思想:
没看答案。
- 统计次数首先想到哈希表。利用unordered_map和multimap的性质,先存储节点的值(key)和其出现的次数(value)于map1中;
- 遍历map1,调换map1中的key-value并添加到map2中(此时出现次数为key,节点值为value);
- 由于众数可能有多个,所以声明num储存某个众数出现的次数,这样遍历map2后出现次数少于众数的节点值就不会被添加到输出数组res中了。
/**
* 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:
unordered_map<int, int> map1;
multimap<int, int, greater<int>> map2;
int num = 0;
void dfs(TreeNode* root){
if(root == nullptr) return;
map1[root->val]++;
dfs(root->left);
dfs(root->right);
}
vector<int> findMode(TreeNode* root) {
vector<int> res;
if(root == nullptr) return res;
dfs(root);
for(auto& v : map1){
map2.insert(pair<int, int>(v.second, v.first));
}
for(auto& ech : map2){
num = max(num, ech.first);
if(ech.first == num){
res.push_back(ech.second);
}
}
return res;
}
};
推荐阅读