在给定单词列表 wordlist 的情况下,我们希望实现一个拼写检查器,将查询单词转换为正确的单词。
对于给定的查询单词 query,拼写检查器将会处理两类拼写错误:
大小写:如果查询匹配单词列表中的某个单词(不区分大小写),则返回的正确单词与单词列表中的大小写相同。
例如:wordlist = ["yellow"], query = "YellOw": correct = "yellow"
例如:wordlist = ["Yellow"], query = "yellow": correct = "Yellow"
例如:wordlist = ["yellow"], query = "yellow": correct = "yellow"
元音错误:如果在将查询单词中的元音(‘a’、‘e’、‘i’、‘o’、‘u’)分别替换为任何元音后,能与单词列表中的单词匹配(不区分大小写),则返回的正确单词与单词列表中的匹配项大小写相同。
例如:wordlist = ["YellOw"], query = "yollow": correct = "YellOw"
例如:wordlist = ["YellOw"], query = "yeellow": correct = "" (无匹配项)
例如:wordlist = ["YellOw"], query = "yllw": correct = "" (无匹配项)
此外,拼写检查器还按照以下优先级规则操作:
当查询完全匹配单词列表中的某个单词(区分大小写)时,应返回相同的单词。
当查询匹配到大小写问题的单词时,您应该返回单词列表中的第一个这样的匹配项。
当查询匹配到元音错误的单词时,您应该返回单词列表中的第一个这样的匹配项。
如果该查询在单词列表中没有匹配项,则应返回空字符串。
给出一些查询 queries,返回一个单词列表 answer,其中 answer[i] 是由查询 query = queries[i] 得到的正确单词。
示例:
输入:wordlist = ["KiTe","kite","hare","Hare"], queries = ["kite","Kite","KiTe","Hare","HARE","Hear","hear","keti","keet","keto"]
输出:["kite","KiTe","KiTe","Hare","hare","","","KiTe","","KiTe"]
提示:
1 <= wordlist.length <= 5000
1 <= queries.length <= 5000
1 <= wordlist[i].length <= 7
1 <= queries[i].length <= 7
wordlist 和 queries 中的所有字符串仅由英文字母组成。
思路:对于查询类问题,我们往往要先考虑采用数据结构,使得时间复杂度尽量最优,本题我们考虑采用一个set和两个map解决问题,其中set主要用来存wordlist原列表,其为了判断所要查询的字符串是否在原列表中存在。剩下两个map的功能是:
1)map1存储wordlist列表中将所有字符串小写后,其不同串对应的最小索引(在字符串变为小写后,可能出现几个串相同的情况,我们要保证其对应的索引最小)。
2)map2用来存储字符串去掉原因字母后生成的字符串对应的最小索引值,为了保证字符串中字符的位置不变,我们将所有元音字母变为特殊字符即可。
class Solution {
public String[] spellchecker(String[] wordlist, String[] queries) {
int n=wordlist.length;
int m=queries.length;
String[] ans=new String[m];
Set<String> st=new HashSet<>();
Map<String,Integer> map1=new HashMap<>();
Map<String,Integer> map2=new HashMap<>();
for(int i=0;i<n;i++) {
st.add(wordlist[i]);
String s1=wordlist[i].toLowerCase();
String s2=change(s1);
if(!map1.containsKey(s1))
map1.put(s1, i);
if(!map2.containsKey(s2))
map2.put(s2, i);
}
for(int i=0;i<m;i++) {
if(st.contains(queries[i])) {
ans[i]=queries[i];
continue;
}
String s1=queries[i].toLowerCase();
String s2=change(s1);
if(map1.containsKey(s1))
ans[i]=wordlist[map1.get(s1)];
else if(map2.containsKey(s2))
ans[i]=wordlist[map2.get(s2)];
else
ans[i]="";
}
return ans;
}
private String change(String s) {
StringBuilder str=new StringBuilder();
for(int i=0;i<s.length();i++) {
if(!check(s.charAt(i)))
str.append(s.charAt(i));
else
str.append("#");
}
return str.toString();
}
private boolean check(char c) {
return (c=='a' || c=='e' || c=='i' || c=='o' || c=='u');
}
}