题目:
给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
计数排序
- HashMap统计出数组中各元素出现的次数
- 建立一个新的数组 s[i],用于存储出现了 i 次的元素有多少个
- 从后往前遍历 s[i] 数组,当元素个数累加到 k 个时停止
- 此时再根据停下时对应的次数在HashMap中找到对应的元素
class Solution {
public int[] topKFrequent(int[] nums, int k) {
//hashMap,1.统计数组中每一个元素出现的次数
Map<Integer,Integer> cnt = new HashMap<>();
for(int num:nums){
cnt.put(num,cnt.getOrDefault(num,0)+1);
}
//2.建立一个数组s[i],记录出现了i次的元素有多少个
int[] s = new int[nums.length+1]; //最多出现nums的元素个数次
for(int i:cnt.values()){
s[i]++;
}
//3.从后往前遍历s[i],当s[i]的值累加到等于k时停下,
//此时大于等于i次的元素即为答案
int n = nums.length;
int m = 0;
while(m!=k){
m+=s[n];
n--;
}
//遍历结束,此时的n多减了一次
//遍历哈希表,元素出现次数大于n次的即为答案
int[] res = new int[k]; //答案数组
int i = 0;
for(int t:cnt.keySet()){
if(cnt.get(t)>n){
res[i++] = t;
}
}
return res;
}
}
注:
Java中map.getOrDeafult(Object key, V defaultValue)方法:
当Map集合中有这个key时,就使用这个key对应的value值;
如果没有就使用默认值defaultValue。
cnt.put(num,cnt.getOrDefault(num,0)+1);
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。