ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[LC 676] Trie树简单运用

2022-07-09 12:41:14  阅读:155  来源: 互联网

标签:return LC idx nums Trie 676 son int str


https://leetcode.cn/problems/implement-magic-dictionary/

static int son[10010][26];
int idx = 0;
static int cnt[10010];
class MagicDictionary {
public:
    //Trie树相关的数据;
    MagicDictionary() {
        memset(son, 0, sizeof son);
        memset(cnt, 0, sizeof cnt);
        idx = 0;
    }

    void insert(string& str)
    {
        int p = 0;
        for (int i = 0; i < str.size(); i++)
        {
            int u = str[i] - 'a';
            if (!son[p][u]) son[p][u] = ++idx;
            p = son[p][u];
        }

        cnt[p]++; //标记字符串结尾;
    }

    void buildDict(vector<string> dictionary) {
        for (auto str : dictionary)
        {
            insert(str);
        }
    }
    
    bool search(string searchWord) {
        return query(searchWord);
    }

    bool query(string & str)
    {
        int nums = 0; //表示在当前Trie树搜索中对应的不同字母个数;
        int p = 0; //进行dfs遍历查找;
        return dfs(str, p,  nums, 0);
    }

    bool dfs(string& str, int p, int nums, int idx)
    {
        if (idx == str.size())
        {
            if (nums == 1 && cnt[p])
                return true; 
            return false;
        }

        //遍历该Trie树上所有分支;
        for (int i = 0; i < 26; i++)
        {
            if (!son[p][i]) continue;
            if (dfs(str, son[p][i], nums + ((str[idx] - 'a') != i) ,idx+1)) return true;
        }
        return false;
    }
};

/**
 * Your MagicDictionary object will be instantiated and called as such:
 * MagicDictionary* obj = new MagicDictionary();
 * obj->buildDict(dictionary);
 * bool param_2 = obj->search(searchWord);
 */

标签:return,LC,idx,nums,Trie,676,son,int,str
来源: https://www.cnblogs.com/zhanghanLeo/p/16460575.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有