0前缀树中等 LeetCode820. 单词的压缩编码

820. 单词的压缩编码

描述

单词数组 words 的 有效编码 由任意助记字符串 s 和下标数组 indices 组成，且满足：

words.length == indices.length
助记字符串 s 以 ‘#’ 字符结尾
对于每个下标 indices[i] ，s 的一个从 indices[i] 开始、到下一个 ‘#’ 字符结束（但不包括 ‘#’）的 子字符串 恰好与 words[i] 相等
给你一个单词数组 words ，返回成功对 words 进行编码的最小助记字符串 s 的长度 。

分析

构建一个前缀树，字符串按照长度由长到短排序，遍历所有的字符串加入到前缀树中，如果加入的时候出现新的结点，意味着是无法借用已经插入的字符串，因此需要另起一个#号来标记。
筛选出需要#标记的所有字符串，给这些字符串长度加1再求总和就是答案。

class Solution {
    public int minimumLengthEncoding(String[] words) {
        Trie trie = new Trie();
        Arrays.sort(words,(a,b)->{
            return b.length() - a.length();
        });
        int ans = 0;
        for (String str : words) {
            ans += trie.merge(str);
        }
        return ans;
    }
}

class Trie {
    TrieNode root;

    Trie() {
        root = new TrieNode();
    }

    int merge(String str) {
        TrieNode cur = root;
        boolean flag = false;
        for (int i = str.length() -1; i >= 0; i--) {
            int index = str.charAt(i) - 'a';
            if (cur.children[index] == null) {
                flag = true;
                cur.children[index] = new TrieNode();
            }
            cur = cur.children[index];
        }
        if (flag) {
            return str.length() + 1;
        }
        return 0;
    }
}

class TrieNode {
    char val;
    TrieNode[] children = new TrieNode[26];

    public TrieNode() {}
}