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(1N100000),代表度熊对于字典的操作次数,接下来

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);
    }
}

    原文作者:Trie树
    原文地址: https://blog.csdn.net/chaiwenjun000/article/details/51459976
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞