347TopKFrequentElements

最新推荐文章于 2024-04-22 11:15:19 发布

原创最新推荐文章于 2024-04-22 11:15:19 发布 · 361 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

64 篇文章

订阅专栏

本文介绍了一种高效算法，通过使用HashMap计数和PriorityQueue实现最小堆来找出给定整数数组中出现频率最高的k个元素。该方法首先统计每个元素的出现次数，然后利用最小堆保留频次最高的k个元素。

题目：已知int数组，输入k，输出出现次数最多的k个元素

思路： 1、HashTable使用同步机制，而HashMap是HashTable的轻量级实现（需要外同步），不允许重复键（即，重复键会覆盖）；2、使用PriorityQueue，先进先出（排序）实现MinHeap，保存k个元素。

public List topKFrequent(int[] nums, int k) {
    	if(k <= 0) return null;
        HashMap ht = new HashMap();
        for(int key : nums){
        	Integer value = ht.get(new Integer(key));
        	if(value == null)
        		ht.put(new Integer(key), new Integer(1));
        	else
        		ht.put(new Integer(key), new Integer(value.intValue() + 1));
        } //O(n) count #element
        PriorityQueue pq = new PriorityQueue(); //achive MinHeap O(nlogn)
        ArrayList list = new ArrayList<>();
        for(Integer key : ht.keySet()){
        	pq.add(new Pair(key, ht.get(key)));
        	if(pq.size() > k){
        		pq.poll();
        	}
        }
       Pair pr;
       while((pr = pq.poll()) != null){
        	list.add(0, pr.getKey());          
        }
        return list;
    }	
    
    class Pair implements Comparable{
    	Integer key;
    	Integer value;
    	
    	
    	
		public Pair(Integer key, Integer value) {
			super();
			this.key = key;
			this.value = value;
		}

		public Integer getKey() {
			return key;
		}

		public void setKey(Integer key) {
			this.key = key;
		}

		public Integer getValue() {
			return value;
		}

		public void setValue(Integer value) {
			this.value = value;
		}

		//make out Pair nature order
		public int compareTo(Pair o) {
			// TODO Auto-generated method stub
			if(value > o.value) 
				return 1;
			else if(value < o.value)
				return -1;
			else
				return 0;
		}
    }