题目地址:
https://www.lintcode.com/problem/unique-word-abbreviation/description
要求设计一个数据结构,给定一个字符串数组对其进行初始化,再给定一个字符串,该数据结构可以查询数组中是否存在某个不同于给定字符串的一个字符串,使得这个字符串的缩写(缩写的方式是,首字母 + 中间字符数量 + 尾字母。若字符串长度小于等于 2 2 2,则缩写等于自己)与给定字符串缩写相同。
思路是用两个哈希表,一个记录每个单词出现的次数,另一个记录每个缩写出现的次数。对于给定的单词,如果发现其出现的次数不等于其缩写出现的次数(由于缩写次数一定大于等于单词次数,如果不等,则一定是大于了),说明有别的单词缩写和它一样,返回false;否则返回true。代码如下:
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
public class ValidWordAbbr {
private Map<String, Integer> dict = new HashMap<>(), abbr = new HashMap<>();
/*
* @param dictionary: a list of words
*/
public ValidWordAbbr(String[] dictionary) {
// do intialization if necessary
for (String s : dictionary) {
dict.put(s, dict.getOrDefault(s, 0) + 1);
String ab = getAbbr(s);
abbr.put(ab, abbr.getOrDefault(ab, 0) + 1);
}
}
/*
* @param word: a string
* @return: true if its abbreviation is unique or false
*/
public boolean isUnique(String word) {
// write your code here
return Objects.equals(dict.get(word), abbr.get(getAbbr(word)));
}
private String getAbbr(String s) {
if (s.length() <= 2) {
return s;
}
return "" + s.charAt(0) + (s.length() - 2) + s.charAt(s.length() - 1);
}
}
初始化时间复杂度 O ( n l ) O(nl) O(nl),查询时间复杂度 O ( l ) O(l) O(l)。空间 O ( n l ) O(nl) O(nl)。