Top K Frequent Elements

bleuesprit

于 2016-05-05 21:45:43 发布

阅读量193

点赞数

CC 4.0 BY-SA版权

分类专栏： leetCode

本文链接：https://blog.youkuaiyun.com/bleuesprit/article/details/51326604

leetCode 专栏收录该内容

199 篇文章

订阅专栏

本文介绍了一种使用Java实现的算法，该算法通过排序和优先队列来找出数组中出现频率最高的k个元素。文章提供了完整的代码示例，包括主函数测试案例以及自定义的记录类用于比较元素频率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接

import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
import java.util.PriorityQueue;

public class Solution {

    public static void main(String args[])
    {
        Solution obj=new Solution();
        List<Integer> result=obj.topKFrequent(new int[]{1,1,1,2,2,3},2);
        for (Integer integer : result) {
            System.out.println(integer);
        }
    }

    public List<Integer> topKFrequent(int[] nums, int k) {
        Arrays.sort(nums);
        PriorityQueue<Record> heap=new PriorityQueue<Record>();
        int last=nums[0];
        int count=0;
        for(int i=0;i<nums.length;i++)
        {
            if(last!=nums[i])
            {
                heap.add(new Record(last, count));
                count=1;
                last=nums[i];
            }
            else
            {
                count++;
            }
        }
        heap.add(new Record(last, count));
        List<Integer> result=new LinkedList<Integer>();
        for(int i=0;i<k;i++)
        {
            result.add(heap.poll().nums);
        }
        return result;
    }
}

class Record implements Comparable<Record>
{
    int nums;
    int count;
    Record(int nums,int count)
    {
        this.nums=nums;
        this.count=count;
    }
    @Override
    public int compareTo(Record o) {

        return count<o.count?1:-1;
    }
}