0
点赞
收藏
分享

微信扫一扫

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;
    }
}
举报

相关推荐

0 条评论