桶排序算法

桶排序

桶排序应用于数组中对元素数量的排序。

主要思路:

  1. 记录每种元素出现的频数,需要遍历一次数组,时间复杂度为O(n)。这里使用哈希来记录,C++中使用unordered_map.并记录出现的频数的数目max_count。
  2. 定义一个bucket二维数组(大小为max_count + 1,bucket[i]代表出现次数为i次),根据第一步记录的unordered_map中的值,依次将unordered_map中的元素放入桶中。循环完成后,bucket[1]中存放的是原始数组中出现了1次的元素,bucket[2]中存放的是原始数组中出现了2次的元素,以此类推。
  3. 根据问题要求,便可以对桶中的元素进行操作。

下面以Leetcode 347 Top k Frequent Elements为例:

	    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> counts;
        int max_count = 0;
        // 统计每个元素出现的频数,并记录共出现了几种频数
        for(const int num : nums){
            max_count = max(max_count,++counts[num]);
        }

        // 定义bucket
        vector<vector<int>> bucket(max_count + 1);
        // 将出现频数为i的元素放到第i个桶中
        for(const auto & p : counts){
            bucket[p.second].push_back(p.first);
        }
        vector<int> res;
        for(int i = max_count;i > 0;i--){
            if(res.size() ==k) break;
            for(const int & num : bucket[i]){
                res.push_back(num);
            }
        }
        return res;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值