思路:前缀树的使用
之前似乎很少写过前缀树,练习下
借鉴一篇博客上的代码,but他的代码是错的啊。。。。 坑了我好久, 坑就是删除的时候要先查询出该前缀有多少串可以匹配,设该值为num,删除过程中中间经过的点要减去这一num,最后一个点V置0,并且next数组全部赋值为-1
#include <iostream>
#include <cstring>
#include <stdio.h>
using namespace std;
struct note
{
int next[26];
int v;
void init()
{
v=0;
memset(next,-1,sizeof(next));
}
};
note l[1200500];
int tot=0;
void add(char s[],int len)
{
int now=0;
for(int i=0;i<len;i++)
{
// cout<<s<<":"<<now<<endl;
int tmp = s[i]-'a';
if(l[now].next[tmp]==-1)
{
int nn=tot++;
l[nn].init();
l[now].next[tmp]=nn;
}
now=l[now].next[tmp];
l[now].v+=1;
}
}
int query(char s[],int len)
{
int now=0;
for(int i=0;i<len;i++)
{
int tmp =s[i]-'a';
if(l[now].next[tmp]==-1)
return 0;
now =l[now].next[tmp];
}
return l[now].v;
}
void delet(char s[],int len,int cnt)
{
int now=0;
for(int i=0;i<len;i++)
{
int tmp=s[i]-'a';
if(l[now].next[tmp]==-1)
return ;
l[now].v-=cnt;
now =l[now].next[tmp];
}
l[now].v=0;
for(int i=0;i<26;i++)
l[now].next[i]=-1;
}
char ss[100],cc[100];
int n;
int main()
{
freopen("in.txt","r",stdin);
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<1200500;i++)
l[i].init();
tot=1;
for(int i=0;i<n;i++)
{
scanf("%s%s",ss,cc);
if(ss[0]=='i')
{
add(cc,strlen(cc));
}
else if(ss[0]=='s')
{
int ans=query(cc,strlen(cc));
if(ans) printf("Yes\n");
else printf("No\n");
}
else
{
int cnt=query(cc,strlen(cc));
delet(cc,strlen(cc),cnt);
}
}
}
}