点击打开链接
Problem C
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 131072/131072 K (Java/Others)Total Submission(s): 2240 Accepted Submission(s): 618
Problem Description
度熊手上有一本神奇的字典,你可以在它里面做如下三个操作:
1、insert : 往神奇字典中插入一个单词
2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词
3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串
1、insert : 往神奇字典中插入一个单词
2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词
3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串
Input
这里仅有一组测试数据。第一行输入一个正整数 N(1≤N≤100000) ,代表度熊对于字典的操作次数,接下来 N 行,每行包含两个字符串,中间中用空格隔开。第一个字符串代表了相关的操作(包括: 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
解题思路:每次插入单词就把相应的字符的出现次数+1,删除的时候先查一遍单词是否出现,若出现了,把前缀的所有字符的出现次数减去最后一个字符的出现次数,再把最后一个字符之后的清零-----字典树
#include<bits/stdc++.h>
using namespace std;
const int maxp=3e6;
char s[50],x[50];
int n,f[maxp][30];
int main()
{int sum=0;memset(f[0],0,sizeof(f[0]));scanf("%d",&n);while(n--){scanf("%s%s",s,x);if(s[0]=='i'){for(int i=0,j=0;x[i];i++){int k=x[i]-'a'+1;if(!f[j][k]){f[j][k]=++sum;memset(f[sum],0,sizeof(f[sum]));}j=f[j][k];f[j][0]++;}}else if(s[0]=='s'){int flag=1;for(int i=0,j=0;x[i];i++){int k=x[i]-'a'+1;j=f[j][k];if(!f[j][0]){flag=0;break;}}if(flag) printf("Yes\n");else printf("No\n");}else{int j=0;for(int i=0;x[i];i++){int k=x[i]-'a'+1;if(!f[j][k]){j=0;break;}elsej=f[j][k];}if(!j)continue;int p=j;j=0;for(int i=0;x[i];i++){int k=x[i]-'a'+1;j=f[j][k];f[j][0]-=f[p][0];}memset(f[p],0,sizeof(f[p]));}}return 0;
}