题目地址:
https://www.lintcode.com/problem/word-abbreviation/description
给定若干小写字母的单词,每个单词长度都大于
1
1
1。要求按照下列规则产生所有单词的缩写:
1、每个单词的缩写格式是其前缀 + 省略的字符个数 + 最后一个字符,但要越短越好;
2、如果多个单词缩写相同,那么要选取最长的前缀,使得这些单词缩写变得不同;
3、如果缩写没有使得单词变得更短,那么不缩写。
要求返回的时候按原数组相同顺序排序。
先将所有字符串缩写成第一个字符 + 数字 + 最后一个字符的形式(当然如果长度不变短那就不缩写),顺便把所有缩写都用哈希表计数。如果有缩写出现了多于一次,那么就尝试一个长度再长一个字符的前缀,直到所有缩写都唯一为止。求缩写的时候,我们可以用一个函数来做,这个函数的参数是字符串以及从哪个位置开始缩写。代码如下:
import java.util.HashMap;
import java.util.Map;
public class Solution {
/**
* @param dict: an array of n distinct non-empty strings
* @return: an array of minimal possible abbreviations for every word
*/
public String[] wordsAbbreviation(String[] dict) {
// write your code here
int len = dict.length;
// 记录最终答案
String[] res = new String[len];
// 记录每个缩写出现了多少次
Map<String, Integer> map = new HashMap<>();
// 记录每个字符是从哪个位置开始缩写的
int[] pref = new int[len];
// 先假设所有字符串s都是从s[1]开始缩写的
for (int i = 0; i < len; i++) {
pref[i] = 1;
res[i] = getAbbr(dict[i], 1);
map.put(res[i], map.getOrDefault(res[i], 0) + 1);
}
// 先假设有重复
boolean unique = false;
while (!unique) {
unique = true;
for (int i = 0; i < len; i++) {
// 如果找到了某个缩写出现次数大于1了,那么就尝试更长的前缀来缩写
if (map.get(res[i]) > 1) {
pref[i]++;
res[i] = getAbbr(dict[i], pref[i]);
map.put(res[i], map.getOrDefault(res[i], 0) + 1);
// 标记为有重复
unique = false;
}
}
}
return res;
}
// 返回s的以s[0 : i - 1]为前缀的缩写,也就是从s[i]开始进行缩写
private String getAbbr(String s, int i) {
// 如果缩写不会使得字符串变短,则直接返回s本身
if (i >= s.length() - 2) {
return s;
}
return s.substring(0, i) + (s.length() - i - 1) + s.charAt(s.length() - 1);
}
}
时空复杂度 O ( n l ) O(nl) O(nl), n n n是单词数量, l l l是单词最长长度。