LeetCode 820: 单词压缩编码 Java实现

题目来源：力扣（LeetCode）820
链接：https://leetcode-cn.com/problems/short-encoding-of-words

给定一个单词列表，我们将这个列表编码成一个索引字符串 S 与一个索引列表 A。

例如，如果这个列表是 [“time”, “me”, “bell”]，我们就可以将其表示为 S = “time#bell#” 和 indexes = [0, 2, 5]。

对于每一个索引，我们可以通过从字符串 S 中索引的位置开始读取字符串，直到 “#” 结束，来恢复我们之前的单词列表。

那么成功对给定单词列表进行编码的最小字符串长度是多少呢？

示例：

输入: words = ["time", "me", "bell"]
输出: 10
说明: S = "time#bell#" ， indexes = [0, 2, 5] 。

提示：

1 <= words.length <= 2000
1 <= words[i].length <= 7
每个单词都是小写字母 。

解题思路
这个题目输入是给定一个String类型的数组，要计算出最小的能完全表示出这些单词的字符串的长度。如果一个单词包含在另外一个单词的后缀之中，就可以将两个单词合并，只用长的单词即可。如"time"和"me"可以合并为"time#".

代码实现

class Solution {
    public int minimumLengthEncoding(String[] words) {
        //排序长的单词排到前面
        Arrays.sort(words, (a, b)->b.length() - a.length());
        String result = words[0];
        for (int i = 1; i < words.length; i++) {
            int index = result.indexOf(words[i]);
            //判断后一个单词是否包含在字符串中 且 是不是单词的后缀
            if (index != -1 && result.substring(index + words[i].length(), index + words[i].length() + 1).equals("#")) {
                continue;
            } else {
                result +=  words[i];
                result += "#";
            }
        }
        //最后返回字符串的长度
        return result.length();
    }
}

使用set集合实现

public static int minimumLengthEncoding(String[] words) {
        Set<String> stringSet = new HashSet<>(Arrays.asList(words));
        for (String word : words) {
            for (int i = 1; i < word.length(); i++) {
                stringSet.remove(word.substring(i));
            }
        }
        int count = 0;
        for (String s : stringSet) {
            count += s.length() + 1;
        }
        return count;
    }