【LeetCode】添加与搜索单词-数据结构设计

💐The Begin💐点点关注，收藏不迷路💐

请你设计一个数据结构，支持 添加新单词 和 查找字符串是否与任何先前添加的字符串匹配 。

实现词典类 WordDictionary ：

WordDictionary() 初始化词典对象
void addWord(word) 将 word 添加到数据结构中，之后可以对它进行匹配
bool search(word) 如果数据结构中存在字符串与 word 匹配，则返回 true ；否则，返回 false 。word 中可能包含一些 ‘.’ ，每个 . 都可以表示任何一个字母。

示例：

输入：
[“WordDictionary”,“addWord”,“addWord”,“addWord”,“search”,“search”,“search”,“search”]
[[],[“bad”],[“dad”],[“mad”],[“pad”],[“bad”],[".ad"],[“b…”]]
输出：
[null,null,null,null,false,true,true,true]

解释：
WordDictionary wordDictionary = new WordDictionary();
wordDictionary.addWord(“bad”);
wordDictionary.addWord(“dad”);
wordDictionary.addWord(“mad”);
wordDictionary.search(“pad”); // 返回 False
wordDictionary.search(“bad”); // 返回 True
wordDictionary.search(".ad"); // 返回 True
wordDictionary.search(“b…”); // 返回 True

提示：

1 <= word.length <= 25
addWord 中的 word 由小写英文字母组成
search 中的 word 由 ‘.’ 或小写英文字母组成
最多调用 104 次 addWord 和 search

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>


#define ALPHABET_SIZE 26


typedef struct TrieNode {
    struct TrieNode* children[ALPHABET_SIZE];
    bool isEndOfWord;
} TrieNode;


typedef struct {
    TrieNode* root;
} WordDictionary;


// 创建新的 Trie 节点
TrieNode* createNode() {
    TrieNode* node = (TrieNode*)malloc(sizeof(TrieNode));
    node->isEndOfWord = false;
    for (int i = 0; i < ALPHABET_SIZE; i++) {
        node->children[i] = NULL;
    }
    return node;
}


// 初始化 WordDictionary
WordDictionary* wordDictionaryCreate() {
    WordDictionary* dict = (WordDictionary*)malloc(sizeof(WordDictionary));
    dict->root = createNode();
    return dict;
}


// 向 WordDictionary 中添加单词
void wordDictionaryAddWord(WordDictionary* obj, char* word) {
    TrieNode* curr = obj->root;
    for (int i = 0; word[i]!= '\0'; i++) {
        int index;
        if (word[i] == '.') {
            index = ALPHABET_SIZE - 1;  // 用 ALPHABET_SIZE - 1 表示通配符 '.'
        } else {
            index = word[i] - 'a';
        }
        if (curr->children[index] == NULL) {
            curr->children[index] = createNode();
        }
        curr = curr->children[index];
    }
    curr->isEndOfWord = true;
}


// 辅助函数，深度优先搜索匹配单词
bool dfs(TrieNode* node, char* word) {
    if (node == NULL) {
        return false;
    }
    if (*word == '\0') {
        return node->isEndOfWord;
    }
    if (*word == '.') {
        for (int i = 0; i < ALPHABET_SIZE; i++) {
            if (dfs(node->children[i], word + 1)) {
                return true;
            }
        }
        return false;
    } else {
        int index = *word - 'a';
        return dfs(node->children[index], word + 1);
    }
}


// 搜索是否存在匹配的单词
bool wordDictionarySearch(WordDictionary* obj, char* word) {
    return dfs(obj->root, word);
}


// 释放 WordDictionary 的内存
void wordDictionaryFree(WordDictionary* obj) {
    // 辅助函数，用于递归释放节点内存
    void freeNodes(TrieNode* node) {
        if (node == NULL) {
            return;
        }
        for (int i = 0; i < ALPHABET_SIZE; i++) {
            freeNodes(node->children[i]);
        }
        free(node);
    }
    freeNodes(obj->root);
    free(obj);
}

💐The End💐点点关注，收藏不迷路💐