题目描述
给定一个段落 (paragraph) 和一个禁用单词列表 (banned)。返回出现次数最多,同时不在禁用列表中的单词。
题目保证至少有一个词不在禁用列表中,而且答案唯一。
禁用列表中的单词用小写字母表示,不含标点符号。段落中的单词不区分大小写。答案都是小写字母。
示例:
输入:
paragraph = "Bob hit a ball, the hit BALL flew far after it was hit."
banned = ["hit"]
输出: "ball"
解释:
"hit" 出现了3次,但它是一个禁用的单词。
"ball" 出现了2次 (同时没有其他单词出现2次),所以它是段落里出现次数最多的,且不在禁用列表中的单词。
注意,所有这些单词在段落里不区分大小写,标点符号需要忽略(即使是紧挨着单词也忽略, 比如 "ball,"),
"hit"不是最终的答案,虽然它出现次数更多,但它在禁用单词列表中。
提示:
- 1 <= 段落长度 <= 1000
- 0 <= 禁用单词个数 <= 100
- 1 <= 禁用单词长度 <= 10
- 答案是唯一的, 且都是小写字母 (即使在 paragraph 里是大写的,即使是一些特定的名词,答案都是小写的。)
- paragraph 只包含字母、空格和下列标点符号!?’,;.
- 不存在没有连字符或者带有连字符的单词。
- 单词里只包含字母,不会出现省略号或者其他标点符号。
答案
我的代码
import java.util.*;
/**
* @Author: Ljx
* @Date: 2022/4/17 8:39
* @role: https://leetcode-cn.com/problems/most-common-word/
*/
public class 最常见的单词 {
public static void main(String[] args) {
System.out.println(mostCommonWord("Bob hit a ball, the hit BALL flew far after it was hit.", new String[]{"hit"}));
}
public static String mostCommonWord(String paragraph, String[] banned) {
// 将banned中的单词添加到set集合中
Set<String> set = new HashSet<>(Arrays.asList(banned));
// 定义Map集合
Map<String,Integer> map = new TreeMap<>();
int start = 0;
// 将paragraph中大写字母转为小写字母
paragraph = paragraph.toLowerCase();
for (int end = 0; end < paragraph.length(); end++) {
// 如果遍历到最后且最后一个字符为字母执行下面操作
if (end==paragraph.length()-1){
if (paragraph.charAt(end)<'z'&¶graph.charAt(end)>'a'){
String substring = paragraph.substring(start);
if (!set.contains(substring)){
map.put(substring,map.getOrDefault(substring,0)+1);
}
}
}
// 遍历到的字符不是字母执行下面操作
if (paragraph.charAt(end)>'z'||paragraph.charAt(end)<'a'){
// 截取字符串
String substring = paragraph.substring(start, end);
// 判断该字符串是否在set集合中
if (!set.contains(substring)){
// 如果不在添加到map集合中
map.put(substring,map.getOrDefault(substring,0)+1);
}
// 将指针后移
while (end<paragraph.length()&&(paragraph.charAt(end)>'z'||paragraph.charAt(end)<'a')){
end++;
start = end;
}
}
}
int x = 0;
String y = "";
// 遍历map集合找到value最大的字符串
for (String key : map.keySet()) {
if (map.get(key)>x){
y = key;
x = map.get(key);
}
}
// 返回字符串
return y;
}
}
官方答案
哈希表 + 计数
为了判断给定段落中的每个单词是否在禁用单词列表中,需要使用哈希集合存储禁用单词列表中的单词。以下将禁用单词列表中的单词称为禁用单词。
遍历段落 paragraph,得到段落中的所有单词,并对每个单词计数,使用哈希表记录每个单词的计数。由于每个单词由连续的字母组成,因此当遇到一个非字母的字符且该字符的前一个字符是字母时,即为一个单词的结束,如果该单词不是禁用单词,则将该单词的计数加 1。如果段落的最后一个字符是字母,则当遍历结束时需要对段落中的最后一个单词判断是否为禁用单词,如果不是禁用单词则将次数加 1。
在遍历段落的过程中,对于每个单词都会更新计数,因此遍历结束之后即可得到最大计数,即出现次数最多的单词的出现次数。
遍历段落之后,遍历哈希表,寻找出现次数等于最大计数的单词,该单词即为最常见的单词。
class Solution {
public String mostCommonWord(String paragraph, String[] banned) {
Set<String> bannedSet = new HashSet<String>();
for (String word : banned) {
bannedSet.add(word);
}
int maxFrequency = 0;
Map<String, Integer> frequencies = new HashMap<String, Integer>();
StringBuffer sb = new StringBuffer();
int length = paragraph.length();
for (int i = 0; i <= length; i++) {
if (i < length && Character.isLetter(paragraph.charAt(i))) {
sb.append(Character.toLowerCase(paragraph.charAt(i)));
} else if (sb.length() > 0) {
String word = sb.toString();
if (!bannedSet.contains(word)) {
int frequency = frequencies.getOrDefault(word, 0) + 1;
frequencies.put(word, frequency);
maxFrequency = Math.max(maxFrequency, frequency);
}
sb.setLength(0);
}
}
String mostCommon = "";
Set<Map.Entry<String, Integer>> entries = frequencies.entrySet();
for (Map.Entry<String, Integer> entry : entries) {
String word = entry.getKey();
int frequency = entry.getValue();
if (frequency == maxFrequency) {
mostCommon = word;
break;
}
}
return mostCommon;
}
}
复杂度分析
-
时间复杂度:O(n + m),其中 n 是段落 paragraph 的长度,m 是禁用单词列表 banned 的长度。遍历禁用单词列表一次将禁用单词存入哈希集合中需要 O(m) 的时间,遍历段落得到每个非禁用单词的计数需要 O(n) 的时间,遍历哈希表得到最常见的单词需要 O(n) 的时间。
-
空间复杂度:O(n + m),其中 n 是段落 paragraph 的长度,m 是禁用单词列表 banned 的长度。存储禁用单词的哈希集合需要 O(m) 的空间,记录每个单词的计数的哈希表需要 O(n) 的空间。