码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode Top K Frequent Elements

时间:2017-02-03 11:00:41      阅读:195      评论:0      收藏:0      [点我收藏+]

标签:time   array   title   lex   map   uniq   cte   允许   ref   

原题链接在这里:https://leetcode.com/problems/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].

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.

题解:

可以使用bucket. 把nums array按照元素出现频率放在bucket中. 最大频率是nums.length, 所以bucket的大小是nums.length+1, 才能对应最大频率.

然后从bucket后扫到前,把nums元素加到res中.

Note: bucket的生成等号右边是ArrayList[] 而不能是ArrayList<Integer>[], 因为complier 不能判准加进ArrayList中的tepe, 只有run time才能知道. 所以这里complier不允许确定type.

Time Complexity: O(n), n = nums.length.

Space: O(n).

AC Java:

 1 public class Solution {
 2     public List<Integer> topKFrequent(int[] nums, int k) {
 3         List<Integer> res = new ArrayList<Integer>();
 4         if(nums == null || nums.length == 0 || k < 1){
 5             return res;
 6         }
 7         
 8         HashMap<Integer, Integer> hm = new HashMap<Integer, Integer>();
 9         for(int num: nums){
10             hm.put(num, hm.getOrDefault(num, 0)+1);
11         }
12         
13         ArrayList<Integer> [] bucket = new ArrayList[nums.length+1];
14         for(int key : hm.keySet()){
15             int freq = hm.get(key);
16             if(bucket[freq] == null){
17                 bucket[freq] = new ArrayList<Integer>();
18             }
19             bucket[freq].add(key);
20         }
21         
22         for(int i = bucket.length-1; i>=0 && res.size()<k ; i--){
23             if(bucket[i] != null){
24                 res.addAll(bucket[i]);
25             }
26         }
27         return res;
28     }
29 }

类似Sort Characters By Frequency.

LeetCode Top K Frequent Elements

标签:time   array   title   lex   map   uniq   cte   允许   ref   

原文地址:http://www.cnblogs.com/Dylan-Java-NYC/p/6362017.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!