首页 > 编程语言 > 详细

Java [Leetcode 347]Top K Frequent Elements

时间:2016-07-31 17:46:21      阅读:226      评论:0      收藏:0      [点我收藏+]

题目描述:

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].

Note: 

    • You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
    • Your algorithm‘s time complexity must be better than O(n log n), where n is the array‘s size.

解题思路:

使用HashMap计数

代码如下:

public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {

    	Map<Integer, Integer> map = new HashMap<Integer, Integer>();
    	List<Integer>[] freqArray = new List[nums.length + 1];

        for(int num : nums){
        	map.put(num, map.getOrDefault(num, 0) + 1);
        }

        for(int key : map.keySet()){
        	int freq = map.get(key);
        	if(freqArray[freq] == null)
        		freqArray[freq] = new ArrayList<Integer> ();
        	freqArray[freq].add(key);
        }

        List<Integer> res = new ArrayList<>();

        for(int i = freqArray.length - 1; i >= 0 & res.size() < k; i--){
        	if(freqArray[i] != null)
        		res.addAll(freqArray[i]);
        }

        return res;

    }
}

  

Java [Leetcode 347]Top K Frequent Elements

原文:http://www.cnblogs.com/zihaowang/p/5723387.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!