Problem C —— 字典树

Problem Description
度熊手上有一本神奇的字典,你可以在它里面做如下三个操作:

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

Source
2016”百度之星” - 资格赛(Astar Round1)

说起来好像都没怎么写过正常的字典树,都是一些奇奇怪怪的水题,这个就有点有趣了,他是递归删边的,注意只有一个字符的情况。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int Map[maxn*30][30],cnt,num[maxn*30][30];
void add(char* s)
{
    int now=0;
    int len=strlen(s);
    for(int i=0;i<len;i++)
    {
        if(Map[now][s[i]-'a']==0)
            Map[now][s[i]-'a']=++cnt;
        num[now][s[i]-'a']++;
        now=Map[now][s[i]-'a'];
    }
}
int del(char* s,int i,int len,int now)
{
    if(Map[now][s[i]-'a']==0)
        return 0;
    if(i==len-1)
    {
        int n=num[now][s[i]-'a'];
        num[now][s[i]-'a']=0;
        Map[now][s[i]-'a']=0;
        return n;
    }
    int ans=del(s,i+1,len,Map[now][s[i]-'a']);
    num[now][s[i]-'a']-=ans;
    if(num[now][s[i]-'a']==0)
        Map[now][s[i]-'a']=0;
    return ans;
}
void query(char* s)
{
    int now=0;
    int len=strlen(s);
    for(int i=0;i<len;i++)
    {
        if(!Map[now][s[i]-'a'])
        {
            printf("No\n");
            return ;
        }
        now=Map[now][s[i]-'a'];
    }
    printf("Yes\n");
}
char s[40];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",&s);
        if(s[0]=='i')
        {
            scanf("%s",s);
            add(s);
        }
        else if(s[0]=='d')
        {
            scanf("%s",s);
            del(s,0,strlen(s),0);
        }
        else
        {
            scanf("%s",s);
            query(s);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/82467505