字典树(Trie树) C++实现



说明:

以下代码是个人按照自己的理解写的,可能有错误或者不太规范的地方,欢迎指出!




代码如下:

//插入、删除、查询、遍历四种操作
//注意:四种操作的函数实现中,T都是指向上一个结点的指针,以此方便操作。
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const double eps = 1e-6;
const int INF = 2e9;
const LL LNF = 9e18;
const int mod = 1e9+7;
const int maxn = 100+10;

typedef struct node
{
    bool exist;     //是否存在单词
    int cnt;        //作为多少个单词的前缀,不算自己
    struct node *next[26];

    void init()
    {
        exist = cnt = 0;
        memset(next,0,sizeof(0));
    }

}Node, *Tree;

void insert(Tree &T, char *s, int k)
{
    if(!T)
    {
        T = new Node;
        T->init();
    }

    if(!s[k]) 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下面是用 C++ 代码实现字典树Trie)的模糊搜索: ```c++ #include <iostream> #include <vector> #include <string> #include <cstring> using namespace std; const int MAX_N = 1000005; const int MAX_M = 10005; const int MAX_L = 25; struct TrieNode { int son[26]; bool is_end; } trie[MAX_N]; int tot = 1; void insert(string s) { int p = 1; for (int i = 0; i < s.length(); i++) { int c = s[i] - 'a'; if (!trie[p].son[c]) { trie[p].son[c] = ++tot; } p = trie[p].son[c]; } trie[p].is_end = true; } vector<string> ans; void dfs(int p, string s, int cnt, int k) { if (trie[p].is_end && cnt <= k) { ans.push_back(s); } if (cnt > k) { return; } for (int i = 0; i < 26; i++) { if (trie[p].son[i]) { char c = 'a' + i; dfs(trie[p].son[i], s + c, cnt + (c != s[s.length() - 1]), k); } } } int main() { int n, m, k; cin >> n >> m >> k; for (int i = 1; i <= n; i++) { string s; cin >> s; insert(s); } for (int i = 1; i <= m; i++) { string s; cin >> s; ans.clear(); dfs(1, "", 0, k); int cnt = 0; for (int j = 0; j < ans.size(); j++) { if (ans[j].find(s) != string::npos) { cnt++; } } cout << cnt << endl; } return 0; } ``` 在这个代码中,`trie` 数组表示字典树,`tot` 表示字典树的节点数量。`insert()` 函数用于将单词插入字典树。`dfs()` 函数用于进行深度优先搜索,其中 `p` 表示当前节点的编号,`s` 表示当前字符串,`cnt` 表示当前编辑距离,`k` 表示最大编辑距离。在 `dfs()` 函数中,如果当前节点是一个单词的结尾,并且编辑距离不超过 `k`,则将当前字符串加入答案数组 `ans` 中。然后遍历当前节点的所有儿子节点,如果某个儿子节点存在,则继续递归搜索。在递归搜索时需要注意,如果当前字符和上一个字符相同,则不需要增加编辑距离,否则需要增加编辑距离。 在主函数中,首先读入字典树中单词的数量 `n`,查询的数量 `m`,以及最大编辑距离 `k`。然后逐个读入查询字符串,对于每个查询字符串,先清空答案数组 `ans`,然后调用 `dfs()` 函数进行搜索。搜索完成后,遍历答案数组 `ans`,如果某个字符串中包含查询字符串,则将计数器 `cnt` 加一。最后输出计数器 `cnt` 即可。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值