LeetCode 前 K 个高频元素

给定一个非空的整数数组,返回其中出现频率前 k 高的元素。

示例 1:

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

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

你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
解法一:

使用 Hashmap,建立数组中元素与其出现次数的映射。然后建立一个拥有 n 个列表的列表 lists,n 代表数组的长度,然后遍历 Hashmap,将出现次数相同的元素保存到同一个列表中,这里的列表为 lists 中下标为出现次数 -1 的列表。最后反向遍历 lists,将出现次数最大的 k 个元素加入 list 中。

    public List<Integer> topKFrequent(int[] nums, int k) {

        HashMap<Integer, Integer> hashMap = new HashMap<>();

        List<List<Integer>> lists = new ArrayList<>();

        List<Integer> list = new ArrayList<>();

        int max = 0;

        for (int i = 0; i < nums.length; i++) {
            if (!hashMap.containsKey(nums[i])) {
                hashMap.put(nums[i], 1);
            } else {
                hashMap.replace(nums[i], hashMap.get(nums[i]) + 1);
            }
        }

        Iterator iterator = hashMap.entrySet().iterator();

        for (int i = 0; i < nums.length; i++) {
            List<Integer> list1 = new ArrayList<>();
            lists.add(list1);
        }

        while (iterator.hasNext()) {
            Map.Entry entry = (Map.Entry) iterator.next();

            max = Math.max(max, (Integer) entry.getValue());

            lists.get((Integer) entry.getValue() - 1).add((Integer) entry.getKey());
        }

        for (int i = lists.size() - 1; i >= 0; i--) {

            for (int j = 0; j < lists.get(i).size(); j++) {
                if (list.size() == k) {
                    break;
                }
                list.add(lists.get(i).get(j));
            }
        }
        return list;
    }
解法二:

我们还可以使用优先队列来做,使用一个最大堆按照映射次数从大到小排列。这里我们需要自己创建一个 Pair 类,因为 Java 自身提供的 Pair 类是不支持元素比较的。我们还需要实现 Comparable 接口,需要注意的是 compareTo 方法: 如果返回值大于 0,则将大的数放后面;如果返回值等于 0,则不做任何处理;如果返回值小于 0,则将小的数放前面。就是说 java 默认的排序方式是升序排序,而我们要的是降序排序。代码如下:

    public List<Integer> topKFrequent(int[] nums, int k) {

        PriorityQueue<Pair> queue = new PriorityQueue<>();

        HashMap<Integer, Integer> hashMap = new HashMap<>();

        List<Integer> list = new ArrayList<>();


        for (int i = 0; i < nums.length; i++) {
            if (!hashMap.containsKey(nums[i])) {
                hashMap.put(nums[i], 1);
            } else {
                hashMap.replace(nums[i], hashMap.get(nums[i]) + 1);
            }
        }

        Iterator iterator = hashMap.entrySet().iterator();

        while (iterator.hasNext()) {
            Map.Entry entry = (Map.Entry) iterator.next();

            queue.add(new Pair((Integer) entry.getKey(), (Integer) entry.getValue()));
        }

        for (int i = 0; i < k; i++) {
            list.add(queue.poll().getKey());
        }

        return list;
    }

    class Pair implements Comparable<Pair> {

        private Integer key;
        private Integer value;

        public Pair(Integer key, Integer value) {
            this.key = key;
            this.value = value;
        }

        public Integer getValue() {
            return value;
        }

        public Integer getKey() {
            return key;
        }

        @Override
        public int compareTo(Pair o) {
            if (this.value > o.getValue()) {
                return -1;
            } else if (this.value < o.getValue()) {
                return 1;
            } else {
                return 0;
            }
        }
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 算法思想贪心思想双指针排序快速选择堆排序桶排序荷兰国旗问题二分查找搜索BFSDFSBacktracking分治动态...
    第六象限阅读 3,272评论 0 0
  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,775评论 0 33
  • leetcode刷题记录本文记录一下leetcode刷题记录,记录一下自己的解法和心得。 LeetCode Two...
    EarthChen阅读 3,510评论 0 6
  • 日本料理都很好看也很好吃,不过那是馆子里的事情。实际家里面能吃到的料理,味道实在是很单调。 总结起来,日料里面常用...
    瞧瞧阅读 1,539评论 1 4
  • 我又辞职了,这是我北漂2年来换的第7份工作,也是做得时间最长的一份工作,8个月的时间。这份工作我有一个不错的头衔,...
    瑾小花呀阅读 225评论 0 3