leetcode：211. 添加与搜索单词 - 数据结构设计

最新推荐文章于 2024-09-24 08:39:28 发布

OceanStar的学习笔记

最新推荐文章于 2024-09-24 08:39:28 发布

阅读量162

点赞数

分类专栏：算法与数据结构文章标签： leetcode

原文链接：https://leetcode-cn.com/problems/design-add-and-search-words-data-structure/

版权

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

1135 篇文章 83 订阅

订阅专栏

该博客主要介绍了LeetCode上的211题，即如何设计一个字典树来实现添加单词和搜索功能。其中，搜索功能因为特殊要求（'.'代表任意字符）而略有复杂，需要遍历所有子节点。博客详细讲解了TrieNode类的实现以及搜索的深度优先搜索策略。

摘要由CSDN通过智能技术生成

题目来源

leetcode：211. 添加与搜索单词 - 数据结构设计

题目描述

在这里插入图片描述

题目解析

这道题是leetcode：208. Implement Trie (Prefix Tree) 实现字典树(前缀树)的衍生，唯一不同的是search需要重写一下，因为这道题里面.可以代替任意字符，所以一旦有了.，就要查找所有的子树，只要有一个返回true，整个search就返回true


class WordDictionary {
    class TrieNode {
    public:
        TrieNode () : children(26, nullptr), is_word(false){

        }
        ~TrieNode (){
            for(auto  it : children){
                delete it;
            }
        }

        std::vector<TrieNode  *> children;
        bool is_word;
    };
public:
    WordDictionary() {
        root = new TrieNode();
    }

    ~WordDictionary(){
        delete root;
    }

    void addWord(string word) {
        TrieNode *p = root;
        for(auto &a : word){
            int i = a - 'a';
            if(p->children[i] == NULL){
                p->children[i] = new TrieNode();
            }
            p = p->children[i];
        }
        p->is_word = true;
    }

    bool search(string word) {
        return dfs(word, root, 0);
    }

    bool dfs(const string & word, TrieNode *p, int index){
        if(index == word.size()){
            return p->is_word;
        }

        if(word[index] == '.'){
            for(auto &it : p->children){
                if(it && dfs(word, it, index + 1)){
                    return true;
                }
            }
            return false;
        }else{
            return p->children[word[index] - 'a'] &&
            dfs(word, p->children[word[index] - 'a'], index + 1);
        }

    }
private:
    TrieNode *root;
};