前K个高频单词(hashmap,priority使用以及自定义排序)

75 篇文章 0 订阅
15 篇文章 0 订阅

在这里插入图片描述
哈希表

class Solution {
    public List<String> topKFrequent(String[] words, int k) {
    HashMap<String,Integer> map=new  HashMap<>();
    for(String ch:words){
        map.put(ch,map.getOrDefault(ch,0)+1);
    }
    List<String> res=new ArrayList<>();
        for(Map.Entry<String,Integer> en:map.entrySet()){
            res.add(en.getKey());
        }
        Collections.sort(res,new Comparator<String>(){
           public int  compare(String a,String b){
               return map.get(a)==map.get(b)?a.compareTo(b):map.get(b)-map.get(a);
           }
        });
           
        
        return res.subList(0, k);
    }
}
class Solution {

public:
    vector<string> topKFrequent(vector<string>& words, int k) {
       vector<string>res;
     unordered_map<string,int> map;
     for(auto ch:words){
         map[ch]++;
         
     }
   for(auto &[key,value]:map){
       res.emplace_back(key);
   }
    sort(res.begin(),res.end(),[&](const string& a,const string& b)->bool{
        return map[a]==map[b]?a<b:map[a]>map[b];
    });
    res.erase(res.begin()+k,res.end());
    return res;
   
    }
};

优先队列(小根堆)

public class Solution {

     public List<String> topKFrequent(String[] words, int k) {
        // 1.先用哈希表统计单词出现的频率
        Map<String, Integer> count = new HashMap();
        for (String word : words) {
            count.put(word, count.getOrDefault(word, 0) + 1);
        }
        // 2.构建小根堆 这里需要自己构建比较规则 此处为 lambda 写法 Java 的优先队列默认实现就是小根堆
        PriorityQueue<String> minHeap = new PriorityQueue<>((s1, s2) -> {
            if (count.get(s1).equals(count.get(s2))) {
                return s2.compareTo(s1);
            } else {
                return count.get(s1) - count.get(s2);
            }
        });
        // 3.依次向堆加入元素。
        for (String s : count.keySet()) {
            minHeap.offer(s);
            // 当堆中元素个数大于 k 个的时候,需要弹出堆顶最小的元素。
            if (minHeap.size() > k) {
                minHeap.poll();
            }
        }
        // 4.依次弹出堆中的 K 个元素,放入结果集合中。
        List<String> res = new ArrayList<String>(k);
        while (minHeap.size() > 0) {
            res.add(minHeap.poll());
        }
        // 5.注意最后需要反转元素的顺序。
        Collections.reverse(res);
        return res;
    }        
}


优先队列大根堆)

class Solution {
    public List<String> topKFrequent(String[] words, int k) {
    HashMap<String,Integer> map=new  HashMap<>();
    for(String ch:words){
        map.put(ch,map.getOrDefault(ch,0)+1);
    }
    PriorityQueue<String> q=new PriorityQueue<String>(new Comparator<String>(){
        public int compare(String a,String b){
            return map.get(a)==map.get(b)?a.compareTo(b):map.get(b)-map.get(a);
        }
    });
    for(String st:map.keySet()){
        q.offer(st);
    }
    List<String> res=new ArrayList<>();
    for(int i=0;i<k;i++){
        res.add(q.poll());
    }
        return res.subList(0, k);
    }
}

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值