您的位置:首页 > 产品设计 > UI/UE

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

2016-06-12 03:43 417 查看
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,该值出现频率作为value。

2. 对于该map,想办法按照value排序。实现方法为: 建立一个list,将每一个map元素,都放入这个list中,并且对list的compare函数进行重写,根据mapvalue值,对list排序(倒序)。

3. 排序后,将list值的前K个元素,保存至结果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. 编写类FEntity , 重写CompareTo函数。

  代码在此:http://blog.csdn.net/yeqiuzs/article/details/51312538

2. 第一步同我一样,接下来进行桶排序。

代码: http://blog.csdn.net/jin_kwok/article/details/51549322
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: