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


[align=left]Problem Description[/align]
度熊手上有一本神奇的字典，你可以在它里面做如下三个操作：

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

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

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

 

[align=left]Input[/align]
这里仅有一组测试数据。第一行输入一个正整数N(1≤N≤100000)，代表度熊对于字典的操作次数，接下来N行，每行包含两个字符串，中间中用空格隔开。第一个字符串代表了相关的操作（包括：
insert, delete 或者 search）。第二个字符串代表了相关操作后指定的那个字符串，第二个字符串的长度不会超过30。第二个字符串仅由小写字母组成。
 

[align=left]Output[/align]
对于每一个search 操作，如果在度熊的字典中存在给定的字符串为前缀的单词，则输出Yes 否则输出 No。
 

[align=left]Sample Input[/align]

5
insert hello
insert hehe
search h
delete he
search hello

 

[align=left]Sample Output[/align]

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