HDU 5687 Problem C

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

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

 sol：Trie的基本操作

插入时，记录下单词中每个字符出现的次数；

删除时，先在Trie中统计以该单词作为前缀的单词个数num，如果num>0，则删除这些单词。删除操作具体为，从根出发，每走到一个结点，该结点出现的次数-num，走到尾结点时，该结点被经过的次数清0，同时，该结点下连接的字符全部清0，即清空子结点。

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
const int maxn = 1e6+5;
const int inf = 0x3f3f3f3f;
struct node
{
    int next[26];
    int val;
}t[maxn];
char f[20], s[50];
int rt = 1;
int num;
  
void insert() 
{
    int u = 0, v;
    int len = strlen(s+1);
    for(int i = 1; i <= len; i++)
    {
        v = s[i] - 'a';
        if (!t[u].next[v]) 
            t[u].next[v] = rt++;
        u = t[u].next[v]; 
        t[u].val++;//这个结点经过的次数+1 
    } 
}
  
void del() 
{
    int u = 0, v;
    int len = strlen(s+1);
     
    for(int i = 1; i <= len; i++)
    {
        v = s[i] - 'a'; 
        u = t[u].next[v]; 
        t[u].val -= num;//删除时，每个结点经过的次数-num
        //可能还有别的单词的长度是小于该前缀的 
    }
    t[u].val = 0;//标记变量清空
          
    for(int j = 0; j < 26; j++) 
    //一定要将其子结点全清空 
        t[u].next[j] = 0;
}
  
int search() 
{
    int u = 0, v;
    int len = strlen(s+1);
     
    for(int i = 1; i <= len; i++)
    {
        v = s[i] - 'a';
        if (!t[u].next[v]) return 0;
        u = t[u].next[v];
        //if (t[u].val == 0) return 0;
    }
    return t[u].val;
}
  
int main() 
{
    
    int n;
    cin >> n;
    memset(t, 0, sizeof(t));
    while(n--){
        scanf("%s%s", f+1, s+1);
        if (f[1] == 'i') 
            insert();
        else
            if (f[1] == 'd') 
             {
                  num = search();  //统计前缀，num为以该串作为前缀的单词数 
                  if (num != 0) 
                     del();//删除这些单词 
            }
        else {
            if (search()) cout << "Yes" << endl;
            else cout << "No" << endl;
        }
    }
    return 0;
}