实现 Trie （前缀树） II

/*
题目思路：
Trie树是一种树形结构，用于高效地存储和检索字符串集合中的键。
主要操作包括插入一个字符串、搜索一个字符串和搜索具有某前缀的字符串。
在Trie树中，每个节点表示字符串中的一个字符，根节点表示空字符。
使用Java Stream的方式来实现Trie树的功能。
*/

import java.util.HashMap;
import java.util.Map;
import java.util.stream.Stream;

class TrieNode {
    public Map<Character, TrieNode> children;
    public boolean isEndOfWord;
    public TrieNode() {
        children = new HashMap<>();
        isEndOfWord = false;
    }
}

class Trie {
    private TrieNode root;

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

    // 插入一个单词到Trie树
    public void insert(String word) {
        TrieNode node = root;
        word.chars().mapToObj(c -> (char) c).forEach(c -> {
            node.children.putIfAbsent(c, new TrieNode());
            node = node.children.get(c);
        });
        node.isEndOfWord = true;
    }

    // 搜索Trie树中是否存在某个单词
    public boolean search(String word) {
        TrieNode node = root;
        return word.chars().mapToObj(c -> (char) c).allMatch(c -> {
            if (node.children.containsKey(c)) {
                node = node.children.get(c);
                return true;
            } else {
                return false;
            }
        }) && node.isEndOfWord;
    }

    // 搜索Trie树中是否存在以某个前缀开头的单词
    public boolean startsWith(String prefix) {
        TrieNode node = root;
        return prefix.chars().mapToObj(c -> (char) c).allMatch(c -> {
            if (node.children.containsKey(c)) {
                node = node.children.get(c);
                return true;
            } else {
                return false;
            }
        });
    }
}