LeetCode 347. 前 K 个高频元素Java

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

 

示例 1:

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

输入: nums = [1], k = 1
输出: [1]
 
提示：

1 <= nums.length <= 105
k 的取值范围是 [1, 数组中不相同的元素的个数]
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
         // 使用hashmap将数组中的元素和该元素出现的次数记录下来
         HashMap<Integer, Integer> map = new HashMap<>();
        int[] res = new int[k];
        for (int num : nums) {
            if (map.containsKey(num)) {
                 map.put(num, map.get(num) + 1);
            } else {
               map.put(num, 1);
            }
        }
         // 出现最多的元素的出现次数
         int maxTimes = 0;   
        // 找出出现次数最多的元素出现的次数
        for(Map.Entry<Integer, Integer> entry : map.entrySet()){
            if(entry.getValue() > maxTimes){
                maxTimes = entry.getValue();
            }
        }

        // 按出现次数从大到小添加到结果数组
        while(k > 0){
            for(Map.Entry<Integer, Integer> entry : map.entrySet()){
                if(entry.getValue() == maxTimes){
                    //此时是将key加入到数组中
                    res[k - 1] = entry.getKey();
                    k--;
                }
            }
            maxTimes--;
        }
        return res;
    }
    
}