Problem C （HDU-5687）（字典树）

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

  1、insert : 往神奇字典中插入一个单词

  2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词

  3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串

Input

这里仅有一组测试数据。第一行输入一个正整数N(1≤N≤100000)N(1≤N≤100000)，代表度熊对于字典的操作次数，接下来NN行，每行包含两个字符串，中间中用空格隔开。第一个字符串代表了相关的操作（包括： 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

题意：中文题，不过多叙述题意。

思路：这道题的话，很明显就是字典树的方法来做，题里所要求的插入和查询就不多说，关键是删除的操作。在进行删除操作的时候，我们先把给定字符串先在字典树中遍历一遍，然后算出这个字符串最后一个字符的出现次数，然后再遍历一遍，每个节点都减去这个次数，将最后节点的子节点全部删除并且删除中间操作即可。

AC代码：

#include <bits/stdc++.h>
typedef long long ll;
const int maxx=500010;
const int maxn=26;
const int inf=0x3f3f3f3f;
using namespace std;
int Trie[maxx][maxn];
int color[maxx];
char s[maxn];
char x[maxn];
int k;
void init()
{
    memset(Trie,0,sizeof(Trie));
    memset(color,0,sizeof(color));
    k=0;
}
void insert(char *w)
{
    int p=0;
    int len=strlen(w);
    for(int i=0; i<len; i++)
    {
        int c=w[i]-'a';
        if(!Trie[p][c])
            Trie[p][c]=++k;
        p=Trie[p][c];
        color[p]++;
    }
}
int search(char *w)
{
    int p=0;
    int len=strlen(w);
    for(int i=0; i<len; i++)
    {
        int c=w[i]-'a';
        if(!Trie[p][c])
            return 0;
        p=Trie[p][c];
    }
    return color[p];
}
void deletea(char *w)
{
    int p=0;
    int ans=0;
    int len=strlen(w);
    for(int i=0; i<len; i++)
    {
        int c=w[i]-'a';
        if(!Trie[p][c])
            return ;
        p=Trie[p][c];
    }
    ans=color[p];
    p=0;
    for(int i=0; i<len; i++)
    {
        int c=w[i]-'a';
        p=Trie[p][c];
        color[p]-=ans;
    }
    for(int i=0; i<=25; i++)
        Trie[p][i]=0;
}
int main()
{
    int n;
    scanf("%d",&n);
    init();
    while(n--)
    {
        scanf("%s",s);
        scanf("%s",x);
        if(s[0]=='i')
            insert(x);
        else if(s[0]=='s')
        {
            int ans=search(x);
            if(!ans)
                printf("No\n");
            else
                printf("Yes\n");
        }
        else
            deletea(x);
    }
    return 0;
}