【347. 前 K 个高频元素】

题目：

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

示例 1:

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

示例 2:

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

计数排序

1. HashMap统计出数组中各元素出现的次数
2. 建立一个新的数组 s[i]，用于存储出现了 i 次的元素有多少个
3. 从后往前遍历 s[i] 数组，当元素个数累加到 k 个时停止
4. 此时再根据停下时对应的次数在HashMap中找到对应的元素

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        //hashMap，1.统计数组中每一个元素出现的次数
        Map<Integer,Integer> cnt = new HashMap<>();
        for(int num:nums){
            cnt.put(num,cnt.getOrDefault(num,0)+1);
        }

        //2.建立一个数组s[i]，记录出现了i次的元素有多少个
        int[] s = new int[nums.length+1];   //最多出现nums的元素个数次
        for(int i:cnt.values()){
            s[i]++;
        }

        //3.从后往前遍历s[i]，当s[i]的值累加到等于k时停下，
        //此时大于等于i次的元素即为答案
        int n = nums.length;
        int m = 0;
        while(m!=k){
            m+=s[n];
            n--;
        }
        //遍历结束，此时的n多减了一次
        //遍历哈希表，元素出现次数大于n次的即为答案
        int[] res = new int[k];     //答案数组
        int i = 0;
        for(int t:cnt.keySet()){
            if(cnt.get(t)>n){
                res[i++] = t;
            }
        }
        return res;
    }
}

注：

Java中map.getOrDeafult(Object key, V defaultValue)方法：
当Map集合中有这个key时，就使用这个key对应的value值；
如果没有就使用默认值defaultValue。
cnt.put(num,cnt.getOrDefault(num,0)+1);

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