算法day17|如何求普通二叉树的众数

算法day17|如何求普通二叉树的众数

501的变式:普通二叉树的众数

如果把二叉搜索树变成普通二叉树,我们该怎么思考呢?这个时候就要回到我一开始的思路了,用哈希表来解决问题。

cpp 复制代码
class Solution {
public:
    unordered_map<int,int> map;
    void traversal(TreeNode* root)
    {
        if(root==nullptr)
        return;
        map[root->val]++;
        traversal(root->left);
        traversal(root->right);
        return;
    }
    bool static cmp(pair<int,int>&a,pair<int,int>&b)
    {
        return a.second>b.second;
    }

    vector<int> findMode(TreeNode* root) {
        vector<int> result;
        if(root==nullptr)
        return result;
        traversal(root);
        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++)
        {
            if(vec[i].second==vec[0].second)
            result.push_back(vec[i].first);
            else
            break;
        }
        return result;
    }
};

总体思路:

  • 首先,用哈希表来记录二叉树的元素及其个数
  • 其次,对哈希表进行排序。由于哈希表的值是不能直接实现排序的,所以我们要**先把哈希表整个存在vector<pair<int,int>>**中,然后用sort函数来对vector排序即可(另外,sort的排序规则是需要单独写的)
  • 排序之后取前面相同的项的键即可

代码细节:

  • 哈希表的插入
cpp 复制代码
map[root->val]++;

这是最简化的形式,map[键],中括号内是键,而map[键]整体就是对应的值,所以可以map["coffee"]=1;或者map[root->val]++;

  • pair<int,int>也是可以作为一种数据类型的,可以代入函数里面,也可以作为vector的数据类型。本质上和int、double没有区别

  • sort函数

cpp 复制代码
  sort(vec.begin(),vec.end(),cmp);

它有三个参数begin, end(区间左闭右开), cmp。其中,cmp参数为排序规则(cmp参数可以不写,如果不写的话,默认从小到大进行排序),如果是想要从大到小的排序,则需要写greater()(即:sort(begin,end,greater()))此处我们需要另写函数,因为数据类型比较特殊。

对于pair<int,int>,函数重写格式:

cpp 复制代码
  bool static cmp(pair<int,int>&a,pair<int,int>&b)
          {
              return a.second>b.second;
          }

函数开头的bool static要注意

  • 如果以vector作为函数返回值的话,不能返回null
相关推荐
程序员-King.4 分钟前
2、桥接模式
c++·桥接模式
chnming19877 分钟前
STL关联式容器之map
开发语言·c++
VertexGeek10 分钟前
Rust学习(八):异常处理和宏编程:
学习·算法·rust
石小石Orz11 分钟前
Three.js + AI:AI 算法生成 3D 萤火虫飞舞效果~
javascript·人工智能·算法
程序伍六七21 分钟前
day16
开发语言·c++
小陈phd38 分钟前
Vscode LinuxC++环境配置
linux·c++·vscode
火山口车神丶1 小时前
某车企ASW面试笔试题
c++·matlab
jiao_mrswang1 小时前
leetcode-18-四数之和
算法·leetcode·职场和发展
qystca1 小时前
洛谷 B3637 最长上升子序列 C语言 记忆化搜索->‘正序‘dp
c语言·开发语言·算法
薯条不要番茄酱1 小时前
数据结构-8.Java. 七大排序算法(中篇)
java·开发语言·数据结构·后端·算法·排序算法·intellij-idea