LeetCode笔记:347. Top K Frequent Elements

问题:

Given a non-empty array of integers, return the k most frequent elements.
For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].
Note:

  • You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
  • Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

大意:

给出一个非空的整型数组,返回最频繁的k个元素。
例子:
给出 [1,1,1,2,2,3] 和 k = 2,返回 [1,2]。
注意:

  • 你可以假设k始终是有效的,1 ≤ k ≤ 不同的元素数。
  • 你的算法时间复杂度要少于O(nlogn),n是数组的尺寸。

思路:

我们要知道哪些数字出现的最多,以及对应的次数,肯定要先遍历一遍并记录下各个数字出现的次数,这里我们用一个HashMap来记录,数字为key,值为它们出现的次数。

接着循环查找出现的最多、次多...的数字,并添加到结果中去。

代码(Java):

public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        
        Map<Integer, String> map = new HashMap<Integer, String>();
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(nums[i])) {// 记录过
                String numStr = map.get(nums[i]);
                int num = Integer.valueOf(numStr).intValue();
                num ++;
                map.put(nums[i], String.valueOf(num));
            } else {// 没记录过
                map.put(nums[i], "1");
            }
        }
        
        int[] keyArr = new int[map.size()];
        int[] valueArr = new int[map.size()];
        int[] used = new int[map.size()];
        int index = 0;
        for (Integer key : map.keySet()) {  
            keyArr[index] = key;
            System.out.println(key);
            valueArr[index] = Integer.valueOf(map.get(key)).intValue();
            System.out.println(map.get(key));
            index ++;
        }  
        
        List<Integer> result = new ArrayList<Integer>();
        while (k > 0) {
            int biggest = 0;
            for (int i = 0; i < used.length; i++) {
                if (used[i] != 1) {
                    biggest = i;
                    break;
                }
            }
            for (int i = 0; i < valueArr.length; i++) {
                if (valueArr[i] > valueArr[biggest] && used[i] != 1) biggest = i;
            }
            result.add(keyArr[biggest]);
            k --;
            used[biggest] = 1;
        }
        
        return result;
        
    }
}

合集:https://github.com/Cloudox/LeetCode-Record


查看作者首页

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,788评论 0 33
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 135,080评论 19 139
  • 一个人跟着团队去旅行,毎次拼房都遇到不同的人,学会与陌生人相处! 刚刚到达美国时,团友中有一位湖南长沙的妹子就想与...
    Mary妹善阅读 776评论 2 1
  • 这是写给初恋的一封信 我要怎么告诉你我还喜欢你,我要怎么放下面子去追回你,我又要怎么去那些你我走过的地方我的生活里...
    玻璃心0508阅读 184评论 0 0
  • Executor框架的最大优点是把任务的提交和执行解耦。要想执行任务,只需把任务描述清楚,提交即可。这个任务是怎么...
    Qi0907阅读 1,920评论 0 1