2016"百度之星" - 资格赛（Astar Round1）C

Problem Description

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

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

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

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

Input

这里仅有一组测试数据。第一行输入一个正整数N (1\leq N\leq 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

Copy

Yes
No
字典树的操作~模版自己加了删除语句

 #include<stdio.h>
 //#include<bits/stdc++.h>
 #include<string.h>
 #include<iostream>
 #include<math.h>
 #include<sstream>
 #include<set>
 #include<queue>
 #include<map>
 #include<vector>
 #include<algorithm>
 #include<limits.h>
 #define inf 0x7fffffff
 #define INFL 0x7fffffffffffffff
 #define lson l,m,rt<<1
 #define rson m+1,r,rt<<1|1
 #define LL long long
 #define ULL unsigned long long
 using namespace std;

 typedef struct Trie{
     int v;
     Trie *next[];
 }Trie;
 Trie root;
 void createTrie(char *str)
 {
     int len = strlen(str);
     Trie *p = &root, *q;
     for(int i=; i<len; ++i)
     {
         int id = str[i]-'a';
         if(p->next[id] == NULL)
         {
             q = (Trie *)malloc(sizeof(root));
             q->v = ;
             for(int j=; j<; ++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=; i<len; ++i)
     {
         int id = str[i]-'a';
         p = p->next[id];
         if(p == NULL)
             return ;
     }
     return p->v;
 }
 void dele(char *str,int n)
 {
     Trie *p = &root;
     int len = strlen(str);
     for(int i=; i<len; ++i)
     {
         int id = str[i]-'a';
         p = p->next[id];
         p->v-=n;
     }
     for(int j=; j<; ++j)
     {
          p->next[j] = NULL;
     }
 }
 int main()
 {
     int t;
     char s1[],s2[];
     for(int i=;i<;i++)
     {
         root.next[i]=NULL;
     }
     cin>>t;
     while(t--)
     {
         scanf("%s%s",s1,s2);
         if(s1[]=='i')
         {
             createTrie(s2);
         }
         else if(s1[]=='s')
         {
             int ans=findTrie(s2);
             if(ans)
             {
                 puts("Yes");
             }
             else
             {
                 puts("No");
             }
         }
         else
         {
             int cnt=findTrie(s2);
             if(cnt)
             {
                 dele(s2,cnt);
             }
         }
     }
     return ;
 }