题目:有一个不为空且仅包含正整数的数组,找出其中出现频率最高的前 K 个数,时间复杂度必须在 O(n log n) 以内。
输入:一行数据包括两部分,一个正整数数组(数字间 ',' 分隔)和一个正整数 K (1 ≤ K ≤ 数组长度),数组和 K 之间有一个空 格。
输出:输出包含前 K 个出现频率最高的数(出现频率相同时,较小的数在前),用 ', ' 分隔,保证升序排列。
输入样例:1,1,1,2,2,3 2
输出样例:1,2
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;
public class frequency {
public static void main(String[] args) {
Scanner in=new Scanner(System.in);
System.out.println(solution(in.nextLine()));
in.close();
}
private static String solution(String line) {
String[] array=line.split(" ");
String[] numbers=array[0].split(",");
int k=Integer.parseInt(array[1]);
List<Integer> num1=new ArrayList<Integer>();
for(int i=0;i<numbers.length;i++){
num1.add(Integer.parseInt(numbers[i]));
}
//提取不重复的数字到list中
List<Integer> list=new ArrayList<Integer>();
list.add(num1.get(0));
int n=0;
int s=0;
for(int j=1;j<num1.size();j++){
s=0;
if(list.get(n)!=num1.get(j)){
for(int m:list){
if(num1.get(j)==m)
s=1;
}
if(s==0){
list.add(num1.get(j));
n++;
}
}
}
Collections.sort(list);
//查询每个数字出现的次数存储在map的value中,以list中的数当key
Map<String,String> map=new HashMap<String,String>();
for(int p:list){
int count=0;
for(int i=0;i<num1.size();i++){
if(p==num1.get(i))
count+=1;
}
map.put(p+"",count+"");
}
List<Map.Entry<String,String>> newmap= mm(map);
String res="";
for(Map.Entry<String,String> mapping:newmap){
if(k==0)
break;
res+=mapping.getKey()+",";
--k;
// System.out.println(mapping.getKey()+":"+mapping.getValue());
}
res = res.substring(0,res.length()-1);
return res;
}
//按照map的value值排序
public static List<Map.Entry<String,String>> mm(Map<String, String> map){
List<Map.Entry<String,String>> list = new ArrayList<Map.Entry<String,String>>(map.entrySet());
//然后通过比较器来实现排序
Collections.sort(list,new Comparator<Map.Entry<String,String>>() {
//升序排序
public int compare(Map.Entry<String, String> o1,
Map.Entry<String, String> o2) {
return o2.getValue().compareTo(o1.getValue());
}
});
return list;
}
}