字典树（单词查找树、Trie树）

2025/4/19 10:11:47 来源：https://blog.csdn.net/m0_73669127/article/details/142766165 浏览: 次关键词：字典树（单词查找树、Trie树）

题目

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int f[N][26], idx, cnt[N];
void insert(char str[])
{int p = 0;for(int i = 0; str[i]; i++){int u = str[i] - 'a';if(!f[p][u]) f[p][u] = ++idx;p = f[p][u];}cnt[p]++;
}
int query(char str[])
{int p = 0;for(int i = 0; str[i]; i++){int u = str[i] - 'a';if(!f[p][u]) return 0;p = f[p][u];}return cnt[p];
}
int main()
{int n;cin >> n;char str[N];for(int i = 1; i <= n; i++){char op;cin >> op >> str;if(op == 'I') insert(str);else cout << query(str) << '\n';}
}