面试算法65:最短的单词编码

题目

输入一个包含n个单词的数组,可以把它们编码成一个字符串和n个下标。例如,单词数组["time","me","bell"]可以编码成一个字符串"time#bell#",然后这些单词就可以通过下标[0,2,5]得到。对于每个下标,都可以从编码得到的字符串中相应的位置开始扫描,直到遇到'#'字符前所经过的子字符串为单词数组中的一个单词。例如,从"time#bell#"下标为2的位置开始扫描,直到遇到'#'前经过子字符串"me"是给定单词数组的第2个单词。给定一个单词数组,请问按照上述规则把这些单词编码之后得到的最短字符串的长度是多少?如果输入的是字符串数组["time","me","bell"],那么编码之后最短的字符串是"time#bell#",长度是10。

分析

这个题目的目标是得到最短的编码,因此,如果一个单词A是另一个单词B的后缀,那么单词A在编码字符串中就不需要单独出现,这是因为单词A可以通过在单词B中偏移下标得到。

前缀树是一种常见的数据结构,它能够很方便地表达一个字符串是另一个字符树串的前缀。这个题目是关于字符串的后缀。要把字符串的后缀转换成前缀也比较直观:如果一个字符串A是另一个字符串B的后缀,分别反转字符串A和B得到A'和B',那么A'是B'的前缀。例如,把字符串"me"和"time"反转分别得到"em"和"emit","em"是"emit"的前缀。

java 复制代码
public class Test {
    public static void main(String[] args) {
        String[] words = {"time", "me", "bell"};
        System.out.println(minimumLengthEncoding(words));
    }

    static class TrieNode {
        public TrieNode[] children;

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

    public static int minimumLengthEncoding(String[] words) {
        TrieNode root = buildTrie(words);
        int[] total = {0};
        // 初始值为1,是因为每个单词后边有一个#号
        dfs(root, 1, total);
        return total[0];
    }

    private static TrieNode buildTrie(String[] words) {
        TrieNode root = new TrieNode();
        for (String word : words) {
            TrieNode node = root;
            for (int i = word.length() - 1; i >= 0; i--) {
                char ch = word.charAt(i);
                if (node.children[ch - 'a'] == null) {
                    node.children[ch - 'a'] = new TrieNode();
                }

                node = node.children[ch - 'a'];
            }
        }

        return root;
    }

    private static void dfs(TrieNode root, int length, int[] total) {
        boolean isLeaf = true;
        for (TrieNode child : root.children) {
            if (child != null) {
                isLeaf = false;
                dfs(child, length + 1, total);
            }
        }

        if (isLeaf) {
            total[0] += length;
        }
    }
}
相关推荐
重庆小透明15 分钟前
力扣刷题记录【1】146.LRU缓存
java·后端·学习·算法·leetcode·缓存
desssq34 分钟前
力扣:70. 爬楼梯
算法·leetcode·职场和发展
金增辉1 小时前
基于C#的OPCServer应用开发,引用WtOPCSvr.dll
c#
clock的时钟1 小时前
暑期数据结构第一天
数据结构·算法
然我1 小时前
别再只用 base64!HTML5 的 Blob 才是二进制处理的王者,面试常考
前端·面试·html
小小小小王王王1 小时前
求猪肉价格最大值
数据结构·c++·算法
莫空00002 小时前
深入理解JavaScript的Reflect API:从原理到实践
前端·面试
磊叔的技术博客2 小时前
LLM 系列(六):模型推理篇
人工智能·面试·llm
岁忧2 小时前
(LeetCode 面试经典 150 题 ) 58. 最后一个单词的长度 (字符串)
java·c++·算法·leetcode·面试·go
BIYing_Aurora2 小时前
【IPMV】图像处理与机器视觉:Lec13 Robust Estimation with RANSAC
图像处理·人工智能·算法·计算机视觉