hdu 5687

Problem C

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

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 <stdio.h>
#include <string.h>

typedef struct node{
	int num;
	node *next[30];
	node(){ 
		memset(next,0,sizeof(next));
		num=0;
	} 
}Trie;
char dir[32],s[32];
void insert(node *root,char *s){
	node *p=root;
	for(int i=0;s[i];i++){
		int x=s[i]-'a';
		if(p->next[x]==NULL)
		p->next[x]=new node;
		p=p->next[x];
		p->num++;
	}
}
int search(node *root,char *s){
	node *p=root;
	for(int i=0;s[i];i++){
		int x=s[i]-'a';
		if(p->next[x]==NULL)
		return 0;
		p=p->next[x];
	}
	return p->num;
}
void Delete(node *root,char *s,int cnt){
	node *p=root;
	for(int i=0;s[i];i++){
		int x=s[i]-'a';
		p=p->next[x];
		p->num-=cnt;
	}
	for(int i=0;i<30;i++)
	p->next[i]=0;
}
int main(){
	int n;
	scanf("%d",&n);
	Trie *root=new node;
	while(n--){
		scanf("%s %s",dir,s);
		if(dir[0]=='i'){
			insert(root,s);
		}
		else if(dir[0]=='s'){
			int ans=search(root,s);
			if(ans)
			printf("Yes\n");
			else
			printf("No\n");
		}
		else{
			int cnt=search(root,s);
			if(cnt)
			Delete(root,s,cnt);
		}
	}
}