LeetCode--No.347--Top K Frequent Elements

Given a non-empty array of integers, return the k most frequent elements.

For example,

Given [1,1,1,2,2,3] and k = 2, return [1,2].

思路:

1. 新建map,将nums中的数据整理放入其中。key为nums值,value为该值出现频率。
2. 想办法对该map的value值进行排序。于是,将map的每一个键值对作为一个整体,放入list中。重新定义该list的compare函数。
3. 将该list的前k个元素的对应key值,存入结果list中,返回。

public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        for(int i = 0;i < nums.length; i++){
            if (map.containsKey(nums[i])){
                map.put(nums[i], map.get(nums[i]) + 1);
            }
            else
                map.put(nums[i], 1);
        }
        List<Map.Entry<Integer, Integer>> list = new ArrayList<Map.Entry<Integer, Integer>>(map.entrySet());
        Collections.sort(list, new Comparator<Map.Entry<Integer, Integer>>(){
            public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2){
                return o2.getValue().compareTo(o1.getValue());
            }
        });
        List<Integer> res = new ArrayList<Integer>();
        for(int i = 0; i < k; i++){
            res.add(i,list.get(i).getKey());
        }
        return res;    
    }
}

关于该题的其他解法:

1. http://blog.csdn.net/yeqiuzs/article/details/51312538
新建class FEntity, 对该类重写CompareTo方法。

2. http://blog.csdn.net/jin_kwok/article/details/51549322
第一步同我的方法一致。之后对map进行木桶排序。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值