2016"百度之星" - 资格赛（Astar Round1）&&HDOJ 5687 Problem C 【字典树】

最新推荐文章于 2019-02-18 21:37:32 发布

原创最新推荐文章于 2019-02-18 21:37:32 发布 · 250 阅读

0 ·

CC 4.0 BY-SA版权

字典树专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个使用Trie树实现的神奇字典，支持插入、删除和搜索操作。特别是在删除操作时，可以删除所有前缀匹配的单词。文章通过示例说明了如何实现这些功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Problem C

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 131072/131072 K (Java/Others)
Total Submission(s): 1222 Accepted Submission(s): 381

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

重点在于“delete”操作

#include <bits/stdc++.h>
using namespace std;
#define mst(a,b) memset((a),(b),sizeof(a))
#define f(i,a,b) for(int i=(a);i<(b);i++)
#define maxn 26
#define mod 9973
#define ll long long
#define rush() int t;scanf("%d",&t);while(t--)
typedef struct Trie
{
    int v;
    Trie *next[maxn];
}Trie;
Trie *root;
void createTrie(char *str)
{
    int len=strlen(str);
    Trie *p=root,*q;
    for(int i=0; i<len; i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)
        {
            q = (Trie *)malloc(sizeof(Trie));
            q->v=1;
            for(int j=0; j<maxn; j++)
                q->next[j]=NULL;
            p->next[id]=q;
            p=p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p=p->next[id];
        }
    }
}
int findTrie(char*str)
{
    int len=strlen(str);
    Trie *p=root;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        p=p->next[id];
        if(p==NULL)
            return 0;
    }
    return p->v;
}
void mydelete(char *str)
{
    Trie *p=root;
    int len=strlen(str);
    int x=findTrie(str);
    f(i,0,len)
    {
        int id=str[i]-'a';
        p=p->next[id];
        p->v-=x;
    }
    f(i,0,maxn)
    p->next[i]=NULL;
}
int main()
{
    char s1[10],s2[35];
    int n;
    scanf("%d",&n);
    root= (Trie *)malloc(sizeof(Trie));
    for(int i=0;i<maxn;i++)
        root->next[i]=NULL;
    while(n--)
    {
        scanf("%s %s",s1,s2);
        if(strcmp(s1,"insert")==0)
            createTrie(s2);
        else if(strcmp(s1,"search")==0)
        {
            if(findTrie(s2))
                printf("Yes\n");
            else printf("No\n");
        }
        else
        {
            if(findTrie(s2))
                mydelete(s2);
        }
    }
    return 0;
}