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

leetcode347. Top K Frequent Elements--java

2018-03-03 23:57 281 查看
题目: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.
给定一个非空数组,返回前k个出现频率最高的元素。
思路:
1. HashMap遍历数组nums[],得到每个元素的数组
2. 将HashMap中的记录,重新按照每个元素出现的次数,存放到list中,即bucket[i]中ki表示元素出现的个数,对应值bucket[ki]为元素值
3. 由于越靠后的元素,其出现的次数越多,因此,从后向前遍历k个非空bucket[],即为所求结果
程序:
class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        
        HashMap<Integer, Integer> record = new HashMap<Integer, Integer>();
        
        for(int i = 0; i < nums.length; i++){
            if(record.containsKey(nums[i]))
                record.put(nums[i], record.get(nums[i])+1);
            else 
                record.put(nums[i],1);
        }
        
         List<Integer>[] bucket=new List[nums.length+1]; 
        for(int key:record.keySet()){
            int count=record.get(key);  
              
            if(bucket[count]==null)  
               bucket[count]=new ArrayList<Integer>();  
            bucket[count].add(key);  
        }  
        
        List<Integer> result=new ArrayList<Integer>();  
        for(int i=nums.length;i>0;i--)  
        {  
            if(bucket[i]!=null&&result.size()<k)  
                result.addAll(bucket[i]);         
        }  
        return result;
    }
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: