前 K 个高频元素--java
给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public int[] topKFrequent(int[] nums, int k) {
int[] result = new int[k];//新建k长度的数组
HashMap<Integer,Integer> map = new HashMap<>();//新建泛型的哈希表
for(int i = 0;i<nums.length;i++){ //用一个哈希表获取每种元素出现的次数
int num = nums[i];//num为key
map.put(num,map.getOrDefault(num,0) + 1);//把每个键的默认值设为0
}
List<Map.Entry<Integer,Integer>> list=new ArrayList<Map.Entry<Integer,Integer>>(map.entrySet());//适合用set集合新建Entry对象的列表
//从大到小排序,更新比较器
Collections.sort(list,(o1,o2)->(o2.getValue()-o1.getValue()));
//输出前k个元素
for(int i=0;i<k;i++){
result[i]=list.get(i).getKey();
}
return result;
}
}