leetcode 347 前 K 个高频元素

思路

原题链接

1. 首先统计数组中各元素出现的次数
2. 在统计出现次数的时候，使用HashMap ，搭配增强for循环，当map中包含当前的键的时候，将其值+1，map.put(num, map.get(num) + 1); 反之，将其值赋值为1 map.put(num, 1);
3. 定义一个最大出现次数maxTimes，使用for循环，统计最大出现次数
4. 使用Map.Entry是为了更方便的输出map键值对，当检测到有值 (即次数) 大于maxTimes的时候，将 entry.getValue() 赋给maxTimes
5. 寻找最大的k个数，步骤如下
6. 在while循环里面使用k>0作为终止条件，使用将map的键值赋给entry，判断，当前的键的值是否是和最大的出现次数相等，若是，则将其加入到结果数组中，然后执行k-- ,然后再进行for循环，寻找是否还有和最大出现次数相等的值，若没有则跳出，将maxTimes–，然后开启新的一轮for循环直到k不满足大于0

class Solution{
    public int[] topKFrequent(int[] nums, int k) {
        int[] res = new int[k];    // 结果数组
        Map<Integer, Integer> map = new HashMap();
        // 统计数组中各元素出现的次数
        for(int num : nums){
            if(map.containsKey(num)){
                map.put(num, map.get(num) + 1);
            }else{
                map.put(num, 1);
            }
        }

        int maxTimes = 0;    // 出现最多的元素的出现次数
        /*
        找出出现次数最多的元素出现的次数
        Map.Entry是为了更方便的输出map键值对。一般情况下，要输出Map中的key 和 value  是先得到key的集合keySet()，然后再迭代（循环）由每个key得到每个value。values()方法是获取集合中的所有值，不包含键，没有对应关系。而Entry可以一次性获得这两个值
         */
        for(Map.Entry<Integer, Integer> entry : map.entrySet()){
            if(entry.getValue() > maxTimes){
                maxTimes = entry.getValue();
            }
        }

        /* 按出现次数从大到小添加到结果数组
        在while循环里面使用k>0作为终止条件，使用将map的键值赋给entry，判断，当前的键的值是否是和最大的出现次数相等的，若是，则将其加入到结果数组中
        然后执行k-- ,然后再进行for循环，寻找是否还有和最大出现次数相等的值，若没有则跳出，将maxTimes--，然后开启新的一轮for循环直到k不满足大于0
        */
        while(k > 0){
            for(Map.Entry<Integer, Integer> entry : map.entrySet()){
                if(entry.getValue() == maxTimes){
                    res[k - 1] = entry.getKey();
                    k--;
                }
            }
            maxTimes--;
        }

        return res;
    }
}