用java实现前缀树

蒋劲豪

已于 2024-03-14 10:29:13 修改

阅读量967

点赞数 33

CC 4.0 BY-SA版权

分类专栏：数据结构和算法文章标签： java 开发语言前缀树

于 2024-03-14 10:27:38 首次发布

;

本文链接：https://blog.youkuaiyun.com/m0_58961367/article/details/136700747

数据结构和算法专栏收录该内容

18 篇文章

订阅专栏

本文介绍了基于TreeNode类的前缀树数据结构，用于高效地插入和搜索字符串。作者提到未来可能改进为使用Map结构以提高适用性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里是用26个字母为搜索结构的，后期可以改为用Map结构的，这样的话适用性更高

/**
 * 前缀树类
 */
static class TreeNode {
    /**
     * pass=1 我和我下面有1个单词
     */
    public int pass;
    /**
     * end=1 1代表结束了
     */
    public int end;
    public TreeNode[] nexts;

    public TreeNode() {
        pass = 0;
        end = 0;
        nexts = new TreeNode[26];//26个字母
    }
}

/**
 * 插入一个单词
 */
private static void insert(TreeNode root, String product_) {
    char[] product = product_.toCharArray();
    TreeNode node = root;
    for (int i = 0; i < product.length; i++) {
        node.pass++;
        int index = product[i] - 'a';
        if (node.nexts[index] == null) {
            node.nexts[index] = new TreeNode();
        }
        node = node.nexts[index];
    }
    node.pass++;
    node.end++;
}