题目描述:
度熊手上有一本神奇的字典,你可以在它里面做如下三个操作:
1、insert : 往神奇字典中插入一个单词
2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词
3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串
输入描述:
这里仅有一组测试数据。第一行输入一个正整数N (1\leq N\leq 100000)N(1≤N≤100000),代表度熊对于字典的操作次数,接下来NN行,每行包含两个字符串,中间中用空格隔开。第一个字符串代表了相关的操作(包括: insert, delete 或者 search)。第二个字符串代表了相关操作后指定的那个字符串,第二个字符串的长度不会超过30。第二个字符串仅由小写字母组成。
输出描述:
对于每一个search 操作,如果在度熊的字典中存在给定的字符串为前缀的单词,则输出Yes 否则输出 No。
样例输入 1:
5
insert hello
insert hehe
search h
delete he
search hello样例输出 1:
Yes
No
#include "bits/stdc++.h"
using namespace std;
int s[500005][36];
int k=1;
int sum[500005];
void insert(string &str){
int p=0;
for(int i=0;i<str.length();i++){
if(!s[p][str[i]-'a'])s[p][str[i]-'a']=k++; //插入
p = s[p][str[i]-'a'];sum[p]++;
}
}
bool search(string &str){
int p=0;
for(int i=0;i<str.length();i++){
if(!s[p][str[i]-'a'])return false; //查找
p=s[p][str[i]-'a'];
}
return sum[p];
}
void del(string &str){//删除
int p=0;
int cnt=0;
for(char i : str){
if(!s[p][i-'a'])return; //如果没有前缀是他的会RE
p = s[p][i-'a'];
}
cnt += sum[p];
p = 0;
for (auto i:str) {
p = s[p][i-'a']; //删除有关他的
sum[p]-=cnt;
}
for (int i = 0; i < 26; i++) { //删除这字符串后链接的 等同于 删除以该字符为前缀的字符串
s[p][i] = 0;
}
}
int main(){
int t;
string s1,s2;
cin>>t;
while (t--){
cin>>s1>>s2;
if(s1=="insert"){
insert(s2);
}else if(s1=="search"){
if(search(s2))cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}else{
del(s2);
}
}
return 0;
}