【Leetcode刷题篇】- Leetcode211 添加与搜索单词-数据结构设计_实现词典类 worddictionary : worddictionary() 初始化词典对象 vo-CSDN博客

本文介绍了一种名为WordDictionary的数据结构的设计与实现方法，该结构支持添加新单词及查找字符串是否能与先前添加的字符串匹配，包括处理带有通配符'.'的查询。

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

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

实现词典类 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”); // return False
wordDictionary.search(“bad”); // return True
wordDictionary.search(".ad"); // return True
wordDictionary.search(“b…”); // return True

class WordDictionary {
    //根据字符串长度分开存放
    Map<Integer,Set<String>> dict = new HashMap<>();
    /** Initialize your data structure here. */
    public WordDictionary() {

    }
    //添加单词
    public void addWord(String word) {
        int length = word.length();
        //判断
        if(dict.get(length)!=null){
            dict.get(length).add(word);
        }else{
            Set<String> set = new HashSet<>();
            set.add(word);
            dict.put(length,set);
        }
    }
    
    // 查找
    public boolean search(String word) {
        Set<String> set = dict.get(word.length());
        if(set==null){
            return false;
        }
        if(set.contains(word)){
            return true;
        }
        // 匹配其余的
        char[] arr = word.toCharArray();
        p:for(String s:set){
            if(arr.length!=s.length()){
                continue;
            }
            // 继续判断
            char[] template = s.toCharArray();
            // 逐个字符对比
            for(int i=0;i<template.length;i++){
                if(arr[i]!='.'&&template[i]!=arr[i]){
                    continue p;
                }
            }
            //添加进来
            set.add(word);
            return true;
        }
        return false;
    }
}

/**
 * Your WordDictionary object will be instantiated and called as such:
 * WordDictionary obj = new WordDictionary();
 * obj.addWord(word);
 * boolean param_2 = obj.search(word);
 */