字典树（java）

最新推荐文章于 2024-04-23 06:32:07 发布

GENNJI.

最新推荐文章于 2024-04-23 06:32:07 发布

阅读量277

点赞数

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/satomiyo/article/details/94588444

版权

数据结构专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍了Trie树的数据结构及其在Java中的实现方法。包括节点的定义、插入、搜索和统计前缀单词数量的功能，并提供了完整的代码示例。Trie树能够高效地处理字符串集合的查找和插入操作，适用于自动补全、拼写检查等应用场景。

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

public class Trie {

    /**
     * 节点信息
     */
    private static class Node {
        Node[] child = new Node[26];
        //该节点被num个单词路过
        int num = 0;
        //该节点上的单词个数
        int wordNum = 0;
        boolean flag = false;
        String val = "";

        public Node() {
        }
    }

    public Trie() {
    }

    public Node root = new Node();

    /**
     * 插入单词
     * @param string
     */
    private void insert(String string) {
        Node p = this.root;
        char[] a = string.toCharArray();
        for (char c : a) {
            int index = c - 'a';
            if (p.child[index] == null) {
                Node q = new Node();
                p.child[index] = q;
            }
            p.child[index].num++;
            p = p.child[index];
        }
        p.flag = true;
        p.val = string;
        p.wordNum++;
    }

    /**
     * 查找单词
     * @param str
     * @return
     */
    private boolean search(String str) {
        Node p = this.root;
        char[] a = str.toCharArray();
        for (char c : a) {
            int index = c - 'a';
            if (p.child[index] == null) {
                return false;
            } else {
                p = p.child[index];
            }
        }
        return p.flag;
    }

    /**
     * 以str为开头的单词个数
     * @param str
     * @return
     */
    private int startWithNum(String str) {
        Node p = this.root;
        char[] a = str.toCharArray();
        for (char c : a) {
            int index = c - 'a';
            if (p.child[index] == null) {
                return 0;
            } else {
                p = p.child[index];
            }
        }
        return p.num;
    }


    /**
     * 按字典序输出
     * @param root
     */
    private static void printPre(Node root) {
        if (root != null) {
            int x = root.wordNum;
            while (x > 0) {
                System.out.println(root.val);
                x--;
            }
            for (Node p : root.child) {
                printPre(p);
            }
        }
    }
    public static void main(String[] args) {
        Trie root = new Trie();
        root.insert("apple");
        root.insert("app");
        root.insert("app");
        System.out.println(root.search("apple"));
        System.out.println(root.search("app"));
        System.out.println(root.search("apl"));
        System.out.println(root.search("sda"));
        System.out.println("----------------------");
        System.out.println(root.startWithNum("app"));
        System.out.println(root.startWithNum("ap"));
        System.out.println(root.startWithNum("appl"));
        System.out.println("---------------");
        //System.out.println(startWith("appp"));

        System.out.println("-----------------");
        printPre(root.root);
    }
}