题目链接: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);
}
}