[LeetCode] 每日一题 Day 20：347. 前 K 个高频元素

题目

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入 : nums = [1], k = 1
输出: [1]

提示：

• 你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
• 你可以按任意顺序返回答案。

解决

思路
用哈希表存储数组里面每个元素出现的次数，但是记录之后题目要求使用O(NlogN)的时间复杂度完成，我一时没想出来好的办法，题解中的堆排序我暂时还没有深入了解，这里先用暴力解决，后期再补上优化的算法

代码

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        for(int num:nums){
            map.put(num,map.getOrDefault(num,0)+1);
        }
        int res[] =new int[k];
        int count=0;
        for(int i=0;i<k;i++){
            int max=0;
            int max_num=0;
            for(int j=0;j<nums.length;j++){
                if(map.get(nums[j])>max){
                    max=map.get(nums[j]);
                    max_num=nums[j];
                }
            }
            res[count]=max_num;
            count++;
            map.put(max_num,0);
        }
        return res;
    }
}