leetcode-820.單詞的壓縮編碼

給定一個單詞列表，我們將這個列表編碼成一個索引字符串 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
每個單詞都是小寫字母 。

跟後綴相關，可以用set存儲，然後遍歷把後綴相同的去掉，最後遍歷set集合，裏面的單詞就是我們要求的所有；

class Solution {
    public int minimumLengthEncoding(String[] words) {
        Set<String> set =  new HashSet<>(Arrays.asList(words));
        int ans = 0;
        for(String  s : words){
            for(int i = 1;i < s.length(); i++){
                set.remove(s.substring(i));
            }
        }

        for(String  s : set){
            ans +=  s.length() + 1;
        }
        return ans;
    }
}

字典樹：

class Solution {
    public int minimumLengthEncoding(String[] words) {
        TrieNode trie = new TrieNode();
        Map<TrieNode, Integer> nodes = new HashMap();

        for(int i = 0; i < words.length; i++){
            String s = words[i];
            TrieNode cur = trie;
            //倒序 插入字典序
            for(int j = s.length()- 1; j >= 0; --j){
                cur = cur.get(s.charAt(j)); 
            }
            nodes.put(cur,i);
        }
        int ans = 0;

        for(TrieNode node : nodes.keySet()){
        // 如果是葉子 節點，才計算長度
            if(node.count == 0){
                ans += words[nodes.get(node)].length() + 1;
            }
        }
        return ans;

    }
}

class TrieNode{
    int count;//記錄是否爲葉子節點
    TrieNode[] children;

    TrieNode(){
        count = 0;
        children = new  TrieNode[26];
    }

    public TrieNode get(char c){
        if(children[c - 'a']  == null){
            children[c - 'a'] = new TrieNode();
            count ++; 
        }
        return children[c - 'a'];
    }
}