Top K Frequent Element

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.
采用HashSet和优先队列(PriorityQueue)可以解决,但是结果却并不是特别好

import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.PriorityQueue;
import java.util.Queue;
public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
         HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
    List<Integer> list = new ArrayList<Integer>();
    for(int i = 0 ; i < nums.length;i++)
    {
        if(map.containsKey(new Integer(nums[i])))
        {
            Integer num = new Integer(map.get(nums[i]));
            map.put(nums[i], ++num);
        }
        else
            map.put(nums[i], new Integer(1));

    }
    Comparator order = new Comparator(){

        @Override
        public int compare(Object o1, Object o2) {
            Entry<Integer, Integer> entry1 = (Entry<Integer, Integer>) o1;
            Entry<Integer, Integer> entry2 = (Entry<Integer, Integer>) o2;
            if(entry1.getValue() > entry2.getValue())  
            {   return -1;  }  
            else if(entry1.getValue()<entry2.getValue())  
            {  return  1;  }  
            else  
            {  return 0;  } 
        }

 };
    Queue priorityQueue = new PriorityQueue(11,order);
    java.util.Iterator<Entry<Integer, Integer>> iter = map.entrySet().iterator();
    while (iter.hasNext()) {
        Entry<Integer, Integer> entry = iter.next();
        priorityQueue.add(entry);
    }
    for(int i = 0 ; i < k ; i++)
    {
        Entry<Integer,Integer> entry = (Entry<Integer, Integer>) priorityQueue.poll();
        list.add(entry.getKey());
    }
    return list;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值