标签：java nums int demo 元素 util 347 new 高频

package leetcode;

import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;
import java.util.Set;

public class demo_347 {
    public int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer, Integer> map=new HashMap<Integer, Integer>();
        int[] res=new int[k];
        //存储数字及其对应的个数
        for (int i = 0; i < nums.length; i++) {
            int num=map.getOrDefault(nums[i], 0);
            map.put(nums[i], num+1);
        }
        //创建一个小顶堆，比较entry中value的值
        Set<Map.Entry<Integer, Integer>> entries=map.entrySet();
        PriorityQueue<Map.Entry<Integer, Integer>> queue=new PriorityQueue<Map.Entry<Integer, Integer>>(new Comparator<Map.Entry<Integer, Integer>>() {
            public int compare(Map.Entry<Integer, Integer> m1,Map.Entry<Integer, Integer> m2) {
                return m1.getValue()-m2.getValue();
            }
        });
        int j=0;
        for (Map.Entry<Integer, Integer> set : entries) {
            if(j<k) {
                //保持大小为k
                queue.offer(set);
                j++;
            }else {
                //加入小顶堆
                if(set.getValue()>queue.peek().getValue()) {
                    queue.poll();
                    queue.offer(set);
                }
            }
        }
        //将栈顶元素的key放入结果集中
        j=0;
        while(queue.size()!=0) {
            res[j]=queue.poll().getKey();
            j=j+1;
        }
        for(int i:res) {
            System.out.println(i);
        }
        return res;
    }
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        demo_347 demo=new demo_347();
        int[] nums= {4,1,-1,2,-1,2,3};
        demo.topKFrequent(nums, 2);
    }

}

标签：java,nums,int,demo,元素,util,347,new,高频
来源： https://www.cnblogs.com/Yshun/p/16215461.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。