Leetcode---前K个高频元素--O(n)复杂度

最新推荐文章于 2021-11-13 00:10:56 发布

原创最新推荐文章于 2021-11-13 00:10:56 发布 · 204 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

算法专栏收录该内容

41 篇文章

订阅专栏

本文详细介绍了如何寻找数组中出现频率最高的前K个元素，提供了两种解决方案：一种是通过排序实现，复杂度为O(nlgn)；另一种是使用HashMap统计频次，再利用数组lists存储相同频次的元素，最后逆序遍历获取结果，效率更高。

前K个高频元素

题目链接：前K个高频元素

思路：

解法一：

首先对整个数组排序，复杂度O(nlgn)
再对每个数字出现的频度排序
最后找出前K个数字

解法二：

将数组遍历存入map集合中，value值存放出现次数
构造一个长度为nums.length+1的lists数组
将频度作为下标添加至lists[i]集合中
最后按标从大到小遍历，获取集合中的值，添加至结果集，加入k个即可

	public List<Integer> topKFrequent(int[] nums, int k) {
		
		Map<Integer,Integer> map = new HashMap<Integer,Integer>();
		for(int num:nums) {
			map.put(num,map.getOrDefault(num, 0)+1);
		}
		//这里加1的原因是不必单独处理下标
		List[] lists = new ArrayList[nums.length+1];
		for(int key:map.keySet()) {
			//取出key值对应的元素个数
			int temp = map.get(key);
			if(lists[temp]==null) {
				lists[temp] = new ArrayList<Integer>();
			}
			//这里就是用list存储的原因，因为同一个value值可能对应着好几个key值
			lists[temp].add(key);
		}
		List<Integer> result = new ArrayList<Integer>();
		for(int i=lists.length-1;result.size()<k&&i>=0;--i) {
			if(lists[i]!=null) {
				result.addAll(lists[i]);
			}
		}
		return result;
    }