面试算法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;
        }
    }
}
相关推荐
mysolisoft几秒前
Avalonia+ReactiveUI实现记录自动更新
c#·avalonia·reactiveui·sourcegenerator
夜月yeyue17 分钟前
个人写HTOS移植shell
c++·mcu·算法·性能优化·架构·mfc
Nix Lockhart25 分钟前
《算法与数据结构》第七章[算法3]:图的最小生成树
c语言·数据结构·算法
UrbanJazzerati28 分钟前
一文看懂指数函数:基础与性质
面试
十重幻想32 分钟前
PTA6-5 使用函数求1到10的阶乘和(C)
java·c语言·算法
名誉寒冰1 小时前
【LeetCode】454. 四数相加 II 【分组+哈希表】详解
算法·leetcode·散列表
心疼你的一切1 小时前
使用Unity引擎开发Rokid主机应用的模型交互操作
游戏·ui·unity·c#·游戏引擎·交互
韩立学长1 小时前
【开题答辩实录分享】以《C#大型超市商品上架调配管理系统的设计与实现》为例进行答辩实录分享
开发语言·c#
crystal_pin1 小时前
前端多端适配与Electron思路
面试
十重幻想1 小时前
PTA6-4 使用函数统计指定数字的个数(C)
c语言·c++·算法