题目:
给一非空的单词列表,返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。
事例:
[“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2 输出: [“i”,
“love”] 解析: “i” 和 “love” 为出现次数最多的两个单词,均为2次。
注意,按字母顺序 “i” 在 “love” 之前。
思路如下:
1.统计每个单词出现的次数,Map来搞定
2.把这些单词放在一个ArrayList中并进行排序,自定制比较规则
2.1在sort方法中,指定一个比较器对象,来描述自定制比较规则
sublist能返回list的一个子区间
class Solution {
public List<String> topKFrequent(String[] words, int k) {
//1.统计每个单词出现的次数,Map来搞定
Map<String,Integer> map = new HashMap<>();
for(String w : words){
int count = map.getOrDefault(w,0);
map.put(w,count + 1);
}
//将map中所有的键都放入一个链表中
List<String> result = new ArrayList<>(map.keySet());
//对这个顺序表进行排序,需要自定制一个比较器来完成我们想要的排序
Collections.sort(result, new MyCompare(map));
return result.subList(0,k);//返回链表的一个子区间,区间(0,K)
}
//比较器
class MyCompare implements Comparator<String>{
private Map<String,Integer> map = new HashMap<>();
public MyCompare(Map<String, Integer> map) {
this.map = map;
}
@Override
public int compare(String o1, String o2) {
if(map.get(o1) == map.get(o2)){
//如果相等,按原有的字符串比较,记住是o1.compareTo(o2)
return o1.compareTo(o2);
}
//如果>0,o2在前面
//如果<0,o2在后面
//所以用o2的value - o1的value
return map.get(o2) - map.get(o1);
}
}
}