347. 前 K 个高频元素

给你一个整数数组 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 个高频元素的集合是唯一的
 

进阶：你所设计算法的时间复杂度 必须 优于 O(n log n) ，其中 n 是数组大小。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

看到题目就想到，直接遍历一边数组，将每个数出现的次数存在其数字对应下标的数组里，再遍历一遍数组，储存出现频率前k高的数字，设置一些值进行比较从而判断出频率最高的k个数字，可以利用数组储存这些数值，起初没想到泛型数组这种东西，想自己写个结构体储存出现频率相同的数字到结构体数组中的，后来了解到一些特殊的容器后重新整理了思路。

利用到的排序方法：桶排序

1、遍历数组，用map储存对应的键值对，数值为key，val为出现的频率

//如果当前数字第一次出现，那么初始化val为1，否则将其val加1
hashMap.put(num, hashMap.getOrDefault(num, 0) + 1);

2、遍历map，根据出现的频率数将其放到频率对应下标的泛型数组中，也就是所谓的桶排序

3、从后往前遍历非空的桶，就可以找到出现频率最高的一些数字。

public int[] topKFrequent(int[] nums, int k) {
        int[] result = new int[k];
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        //创建一个泛型数组,数组的大小考虑到最坏的情况，即数组中的数字均为相同的数字
        //那么就会有nums.length+1个桶,因为数组的下标是从0开始的，所以要加1
        //将出现频率为i的数插入到对应下表为i的泛型数组中
        List<Integer>[] list = new ArrayList[nums.length + 1];

        for (int num : nums) {
            //如果当前数字第一次出现，那么初始化val为1，否则将其val加1
            hashMap.put(num, hashMap.getOrDefault(num, 0) + 1);
        }

        //接下来遍历hashMap，以其出现的频率作为下标，其数值作为值插入到泛型数组list中
        for (int num : hashMap.keySet()) {
            int rate = hashMap.get(num);
            if (list[rate] == null) {
                list[rate] = new ArrayList<>();
            }
            list[rate].add(num);
        }
        int t = 0;
        for (int i = nums.length; i > 0; i--) {
            if (list[i] != null) {
                for (int j = 0; j < list[i].size(); j++) {
                    result[t++] = list[i].get(j);
                    if (t == k) {
                        return result;
                    }
                }
            }
        }
        return result;
    }