题目描述
维护一个字符串集合,支持两种操作:
I x
向集合中插入一个字符串 x;Q x
询问一个字符串在集合中出现了多少次。
共有 N 个操作,输入的字符串总长度不超过 1 0 5 10^5 105,字符串仅包含小写英文字母。
输入格式
第一行包含整数 N,表示操作数。
接下来 N 行,每行包含一个操作指令,指令为 I x
或 Q x
中的一种。
输出格式
对于每个询问指令 Q x
,都要输出一个整数作为结果,表示 x 在集合中出现的次数。
每个结果占一行。
数据范围
1 ≤ N ≤ 2 ∗ 1 0 4 1≤N≤2∗10^4 1≤N≤2∗104
输入样例:
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例:
1
0
1
#include<bits/stdc++.h>
#include <unordered_map>
using namespace std;
template <typename T> void debug(string s, T x) { cout << s << "=" << x << "\n"; }
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
typedef pair<int, int>pii;
const ll N = 2e5 + 5;
const ll MOD = 1e4;
const ll INF = 0x7fffffff;
ll son[N][26]; //数组形式定义字典树,值存储的是下一个字符的位置
ll num[N]; //以某一字符串为前缀的单词的数量
ll pos = 1; //当前新分配的存储位置
void insert(string word) {
ll p = 0;
for (int i = 0; word[i]; i++) {
ll n = word[i] - 'a';
if (son[p][n] == 0)son[p][n] = ++pos;//如果对应字符还没有值
p = son[p][n];
}
num[p]++;//存储以当前字母结尾的点的个数
}
ll find(string word) {
ll p = 0;
for (ll i = 0; word[i]; i++) {
ll n = word[i] - 'a';
if (son[p][n] == 0)return 0;//没有走到此字符串的结尾说明没有
p = son[p][n];
}
return num[p];//最后返回的是此字符串出现的次数
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
int t;
cin >> t;
while (t--) {
string s,t;
cin >> s >> t;
if (s == "I") insert(t);
else cout << find(t) << "\n";
}
return 0;
}