给一非空的单词列表,返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。
题目来源:前K个高频单词
解题思路:
static class MyComparator implements Comparator<String>{
private Map<String,Integer> map;
public MyComparator(Map<String,Integer> map){
this.map=map;
}
@Override
public int compare(String o1, String o2) {
int count1=map.get(o1);
int count2=map.get(o2);
if(o1==o2){
//String自身也实现了Comparable,自带字典序的比较功能的
//compareTo就是在使用String默认的比较规则
return o1.compareTo(o2);
}
//o1<o2返回<0
//o1>o2返回>0
//o1==o2返回0
//count1-count2升序排序
//count2-count1降序排序
//sort就是按照升序来排序,小的在前,大的在后
//count1-count2就是定义了出现次数比较少的比较小
//count2-count1就是定义了出现次数比较大的比较小
return count2-count1;
}
}
public List<String> topKFrequent(String[] words, int k){
//1.先统计每个单词出现的次数
Map<String,Integer> map=new HashMap<>();
for(String s:words){
Integer count=map.getOrDefault(s,0);
map.put(s,count+1);
}
//2.把刚才这里的统计得到的字符串内容放到ArrayList中
//keySet相当于得到了一个Set,Set中存放的就是所有的key
ArrayList<String> arrayList=new ArrayList(map.keySet());
//3.按照刚才字符串出现次数,针对ArrayList进行排序
//sort默认是按照元素的自身大小进行升序排序(String的字典序)
//此处我们需要按照字符串出现次数来降序排序
Collections.sort(arrayList,new MyComparator(map));
return arrayList.subList(0,k);
}
在上面的思路中除了在外部创建构造器,也可以在类的内部使用匿名内部类来完成比较
//另一种思路,匿名内部类
//不知道类名字是啥,但是知道这个类实现了Comparator接口
//这个类只需要使用一次,使用完就丢了
Collections.sort(list,new Comparator<String>(){
@Override
public int compare(String o1,String o2){
int count1=map.get(o1);
int count2=map.get(o2);
if(count1==count2){
return o1.compareTo(o2);
}
return count2-count1;
}
});