【Lintcode】639. Word Abbreviation

题目地址：

https://www.lintcode.com/problem/word-abbreviation/description

给定若干小写字母的单词，每个单词长度都大于$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(nl)$，$n$是单词数量，$l$是单词最长长度。