题目:给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
说明:
你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
题解:
将数组值作为key,出现的频次作为value,遍历数组,构建HashMap,然后根据map的value进行排序,之后遍历排序好的map,将前k个出现频次最高的Key返回.
1.构建map,map.getOrDefault(x,0)会对首次key的value值进行默认值赋值
Map map = new HashMap();
for(int x:nums) {
map.put(x, map.getOrDefault(x,0)+1);
}
2.对map根据value值进行排序,Collections.sort第一个参数需要传入list类型,所以将map中的键值对封装为list传入,o2.getValue()-o1.getValue()表示降序排序
List> list = new ArrayList>(map.entrySet());
Collections.sort(list, new Comparator>() {
@Override
public int compare(Entry o1, Entry o2) {
// TODO 自动生成的方法存根
return o2.getValue()-o1.getValue();
}
});
3.从排好序的map中返回前k个出现频次最高的key
注意:因为sort方法是对List中的键值对进行排序的,所以我们需要遍历List才可以保证我们访问的排序顺序。
if(list.size()>=k) {
for(Map.Entry set:list.subList(0, k)) {
ret.add(set.getKey());
}
}
else {
for(Map.Entry set:list) {
ret.add(set.getKey());
}
}
整体算法:
package com.ljq.sort;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;
/**
* @author 作者 Your-Name:
* @version 创建时间:2019年8月27日 下午3:21:23
* 类说明
*/
public class topK {
public static void main(String[] args) {
// TODO 自动生成的方法存根
Scanner scanner = new Scanner(System.in);
while(scanner.hasNext()) {
String[] strings = scanner.nextLine().split(" ");
int len = strings.length;
int k = Integer.parseInt(strings[0]);
int[] nums = new int[len-1];
for(int i=0;i ret = topKFrequent(nums,k);
for(int x:ret) {
System.out.print(x+" ");
}
System.out.println();
}
scanner.close();
}
public static List topKFrequent(int[] nums,int k){
List ret = new ArrayList();
if(k>nums.length) {
return ret;
}
Map map = new HashMap();
for(int x:nums) {
map.put(x, map.getOrDefault(x,0)+1);
}
List> list = new ArrayList>(map.entrySet());
Collections.sort(list, new Comparator>() {
@Override
public int compare(Entry o1, Entry o2) {
// TODO 自动生成的方法存根
return o2.getValue()-o1.getValue();
}
});
if(list.size()>=k) {
for(Map.Entry set:list.subList(0, k)) {
ret.add(set.getKey());
}
}else {
for(Map.Entry set:list) {
ret.add(set.getKey());
}
}
return ret;
}
}