zoukankan      html  css  js  c++  java
  • 【LeetCode】347-前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 是数组的大小。

    解题思路

    采用桶排序。所谓桶排序就是设置若干个带下标的桶,每个桶的下标就是桶内元素出现的次数。

    想要找到 topk,只需要从后往前数 k 个元素出来就可以了。

    Java 实现

    public List<Integer> topKFrequent (int[] nums, int k) {
    
        Map<Integer, Integer> frequencyMap = new HashMap<Integer, Integer>();
        for (int num : nums) {
            frequencyMap.put(num, frequencyMap.getOrDefault(num, 0) + 1);
        }
    
        List<Integer>[] bucket = new List[nums.length + 1];
        for (Integer key : frequencyMap.keySet()) {
            int freq = frequencyMap.get(key);
            if (bucket[freq] == null) {
                bucket[freq] = new ArrayList<>();
            }
            bucket[freq].add(key);
        }
    
        List<Integer> ans = new ArrayList<>();
    
        for (int pos = bucket.length - 1; pos >= 0 && ans.size() < k; pos--) {
            if (bucket[pos] != null) {
                ans.addAll(bucket[pos]);
            }
        }
    
        return ans;
    }
    

    心得体会

    使用HashMapgetOrDefault()构造每个元素出现频率的集合。

  • 相关阅读:
    HDU 3401 Trade
    POJ 1151 Atlantis
    HDU 3415 Max Sum of MaxKsubsequence
    HDU 4234 Moving Points
    HDU 4258 Covered Walkway
    HDU 4391 Paint The Wall
    HDU 1199 Color the Ball
    HDU 4374 One hundred layer
    HDU 3507 Print Article
    GCC特性之__init修饰解析 kasalyn的专栏 博客频道 CSDN.NET
  • 原文地址:https://www.cnblogs.com/yuzhenzero/p/10696920.html
Copyright © 2011-2022 走看看