leetcode 347. Top K Frequent Elements 解题报告

最新推荐文章于 2022-05-22 12:23:46 发布

jly0612

最新推荐文章于 2022-05-22 12:23:46 发布

阅读量341

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/jly0612/article/details/51759817

版权

leetcode 专栏收录该内容

37 篇文章 0 订阅

订阅专栏

本文介绍了一种寻找数组中最常出现的Top K元素的算法实现。通过使用HashMap来记录每个元素出现的频率，并借助TreeMap进行排序，最终筛选出出现频率最高的K个元素。此外还提到了一种改进方案，用于更高效地解决问题。

摘要由CSDN通过智能技术生成

题目链接

解题思路

使用Map将数字和对应的频率按照键值对存储起来
使用TreeMap和HashMap进行利用value值进行排序
按照题目要求输出前K大

使用的方法不够好。后面又使用了另外的方法另建一个对象存储数字和对应的频率，然后将此对象作为value存储在map和list中，对list进行排序。代码就不贴了。

解题代码

public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();  
        ValueComparator bvc =  new ValueComparator(map); 
        TreeMap<Integer,Integer> tmap = new TreeMap<Integer,Integer>(bvc);

        List<Integer> list = new ArrayList<Integer>();
        for(int i = 0;i<nums.length;i++){
            if(map.containsKey(nums[i])){
                map.replace(nums[i],map.get(nums[i])+1);
            }else{
                map.put(nums[i],1);
            }
        }
        tmap.putAll(map);  
        Set set = tmap.keySet();
        Iterator it = set.iterator();
        while(it.hasNext() && k-- >0){
            list.add((Integer) it.next());
        }
        return list;
    }
    class ValueComparator implements Comparator<Integer> {
        Map<Integer,Integer> base;
        public ValueComparator(Map<Integer,Integer> base) {  
            this.base = base;  
        }
        public int compare(Integer a, Integer b) {  
            if(base.get(a)>=base.get(b)){
                return -1;
            }else{
                return 1;
            }     
        } 
    }
}