码迷,mamicode.com
首页 > 编程语言 > 详细

Java [Leetcode 347]Top K Frequent Elements

时间:2016-07-31 17:47:53      阅读:199      评论: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
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!