2016"百度之星" - 资格赛 Problem C 字典树

Problem C
Accepts: 832
Submissions: 6612
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 131072/131072 K (Java/Others)
Problem Description

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

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

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

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

Input

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

用Trie树模拟三种操作即可：

#include<iostream>#include<cstring>#include<vector>#include<cstdio>using namespace std;const int maxnode=1000000+10;struct Trie{    int ch[maxnode][27];    int val[maxnode];    int sz;    void clear(){sz=1;memset(ch[0],0,sizeof(ch[0]));}    int idx(char c){return c-'a';}    void insert(const char *s)    {        int u=0,n=strlen(s);        for(int i=0;i<n;i++){            int c=idx(s[i]);            if(!ch[u][c]){                memset(ch[sz],0,sizeof(ch[sz]));                val[sz]=1;                ch[u][c]=sz++;            }            else val[ch[u][c]]++;            u=ch[u][c];        }    }    bool search(const char *s)    {        int u=0,len=strlen(s);        for(int i=0;i<len;i++){            int c=idx(s[i]);            //cout<<val[ch[u][c]]<<endl;            if(!ch[u][c]||!val[ch[u][c]])return 0;            u=ch[u][c];        }        return 1;    }    void del(const char *s)    {        if(!this->search(s))return;        int u=0,len=strlen(s);        for(int i=0;i<len;i++){            int c=idx(s[i]);            u=ch[u][c];           // cout<<val[u]<<endl;        }        memset(ch[u],0,sizeof(ch[u]));        int x=val[u];        u=0;        for(int i=0;i<len;i++){            int c=idx(s[i]);            u=ch[u][c];            val[u]-=x;        }    }};char s[33],op[23];Trie trie;int main(){    //freopen("f.txt","r",stdin);    int T;    scanf("%d",&T);    trie.clear();    while(T--){        scanf("%s%s",op,s);        if(op[0]=='i'){            trie.insert(s);        }        else if(op[0]=='s'){            if(trie.search(s))puts("Yes");            else puts("No");        }        else{            trie.del(s);        }    }    return 0;}