HDU5687（字典树Trie增删查）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5687

Problem C

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

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
  
  

注意删除操作，删除的是前缀相同的所有单词

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <set>
#define mem(a,x) memset(a,x,sizeof(a))
struct Node{
    Node *child[26];
    int cnt;
    Node(){
        mem(child,NULL); cnt = 0;
    }
}*root;
void Insert(char *s){
    int len = strlen(s);
    Node *p = root;
    for(int i=0;i<len;i++){
        int k = s[i] - 'a';
        if(p->child[k] == NULL){
            p->child[k] = new Node();
        }
        p = p->child[k];
        p->cnt++;
    }
}
void DeleteNode(Node *x){
    if(x == NULL) return;
    for(int i=0;i<26;i++){
        DeleteNode(x->child[i]);
    }
    delete x;
}
int Find(char *s){
    int len = strlen(s);
    Node *p = root;
    for(int i=0;i<len;i++){
        int k = s[i] - 'a';
        if(p->child[k] == NULL) return 0;
        p = p->child[k];
    }
    return p->cnt;
}
void Delete(char *s){
    int len = strlen(s);
    Node *p = root;
    Node *q = NULL;
    for(int i=0;i<len;i++){
        int k = s[i] - 'a';
        if(p->child[k] == NULL) return ;
        p = p->child[k];
    }
    int num = p->cnt; // 找出要删除单词的个数
    p = root;
    for(int i=0;i<len;i++){
        int k = s[i] - 'a';
        q = p->child[k]; //①注意这里删除操作需要用另一个指针节点，因为删除之后就找不到当前节点了
        q->cnt -= num; //每一个节点的cnt 都要减去num;
        if(q->cnt == 0){
            DeleteNode(q);
            p->child[k] = NULL; //②通过p还可以找到当前节点，然后置为空
            return ;
        }
        p = p->child[k];
    }
}
int main(){
    char op[10], word[35];
    int n;
    while(~scanf("%d", &n)){
        root = new Node();
        for(int i = 0; i < n; ++i){
            scanf("%s%s", op, word);
            if(op[0] == 'i'){
                Insert(word);
            } else if(op[0] == 'd'){
                Delete(word);
            } else {
                if(Find(word)) printf("Yes\n");
                else printf("No\n");
            }
        }
        DeleteNode(root);
    }
}