Problem C
Accepts: 832
Submissions: 6612
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 131072/131072 K (Java/Others)
Problem Description
度熊手上有一本神奇的字典,你可以在它里面做如下三个操作:
1、insert : 往神奇字典中插入一个单词
2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词
3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串
Input
这里仅有一组测试数据。第一行输入一个正整数N(1≤N≤100000)N (1leq Nleq 100000)N(1≤N≤100000),代表度熊对于字典的操作次数,接下来NNN行,每行包含两个字符串,中间中用空格隔开。第一个字符串代表了相关的操作(包括: insert, delete 或者 search)。第二个字符串代表了相关操作后指定的那个字符串,第二个字符串的长度不会超过30。第二个字符串仅由小写字母组成。
Output
对于每一个search 操作,如果在度熊的字典中存在给定的字符串为前缀的单词,则输出Yes 否则输出 No。
Sample Input
5
insert hello
insert hehe
search h
delete he
search hello
Sample Output
Yes
No
用Trie树模拟三种操作即可:
#include<iostream>
#include<cstring>
#include<vector>
#include<cstdio>
using namespace std;
const int maxnode=1000000+10;
struct Trie
{
int ch[maxnode][27];
int val[maxnode];
int sz;
void clear(){sz=1;memset(ch[0],0,sizeof(ch[0]));}
int idx(char c){return c-'a';}
void insert(const char *s)
{
int u=0,n=strlen(s);
for(int i=0;i<n;i++){
int c=idx(s[i]);
if(!ch[u][c]){
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=1;
ch[u][c]=sz++;
}
else val[ch[u][c]]++;
u=ch[u][c];
}
}
bool search(const char *s)
{
int u=0,len=strlen(s);
for(int i=0;i<len;i++){
int c=idx(s[i]);
//cout<<val[ch[u][c]]<<endl;
if(!ch[u][c]||!val[ch[u][c]])return 0;
u=ch[u][c];
}
return 1;
}
void del(const char *s)
{
if(!this->search(s))return;
int u=0,len=strlen(s);
for(int i=0;i<len;i++){
int c=idx(s[i]);
u=ch[u][c];
// cout<<val[u]<<endl;
}
memset(ch[u],0,sizeof(ch[u]));
int x=val[u];
u=0;
for(int i=0;i<len;i++){
int c=idx(s[i]);
u=ch[u][c];
val[u]-=x;
}
}
};
char s[33],op[23];
Trie trie;
int main()
{
//freopen("f.txt","r",stdin);
int T;
scanf("%d",&T);
trie.clear();
while(T--){
scanf("%s%s",op,s);
if(op[0]=='i'){
trie.insert(s);
}
else if(op[0]=='s'){
if(trie.search(s))puts("Yes");
else puts("No");
}
else{
trie.del(s);
}
}
return 0;
}