zoukankan      html  css  js  c++  java
  • leetcode 前K个高频元素

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

    示例 1:

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

    示例 2:

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

    说明:

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

    ------

    思路:

      首先利用hashMap,通过遍历一边,存储元素和相应的次数 。 hashMap<元素,次数>

      其次,利用一个数组,长度为n+1,数组下标对应出现的次数, 遍历hashmap,将出现 i次的元素,存储在 数组[i]的位置上,感觉类似于倒排索引。

      最后,倒序查找这个数组,取k个,即为结果。

    public List<Integer> topKFrequent(int[] nums, int k) {
            if (nums == null || nums.length == 0) {
                return null;
            }
            HashMap<Integer, Integer> hashMap = new HashMap<>();
            for (int i = 0; i < nums.length; i++) {
                if (hashMap.containsKey(nums[i])) {
                    hashMap.put(nums[i], hashMap.get(nums[i]) + 1);
                } else {
                    hashMap.put(nums[i], 1);
                }
            }
          ArrayList
    <Integer>[] list = new ArrayList[nums.length + 1]; for (int key : hashMap.keySet()) { int value = hashMap.get(key);        //以value 为下标,将key存储 list[] 数组, list[] 中是一个 arrayList if (list[value] == null) { ArrayList<Integer> temp = new ArrayList<>(); temp.add(key); list[value] = temp; } else { list[value].add(key); } } ArrayList<Integer> res = new ArrayList<>(); //存放结果 for (int i = list.length - 1; i >= 0 && res.size() < k; i--) { if (list[i] != null) { res.addAll(list[i]); } } return res; }
  • 相关阅读:
    [51nod] 1301 集合异或和
    [BZOJ] 1088: [SCOI2005]扫雷Mine
    [LUOGU] P4251 [SCOI2015]小凸玩矩阵
    8.21模拟赛
    [BZOJ] 3163: [Heoi2013]Eden的新背包问题
    [BZOJ] 1001: [BeiJing2006]狼抓兔子
    【NOIP2017提高A组冲刺11.8】好文章
    [BZOJ] 1520: [POI2006]Szk-Schools
    [BZOJ] 1877: [SDOI2009]晨跑
    day23(事务管理)
  • 原文地址:https://www.cnblogs.com/magicya/p/10096451.html
Copyright © 2011-2022 走看看