hdu5687

Problem C

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

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<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
struct Trie
{
	Trie*next[26];   //子节点 
	Trie*fa;          //父节点 
	int flag;  //等否成为前缀 
};
Trie*root;
void init()         //初始化 
{
	root=new Trie;
	for(int i=0;i<26;i++)
		root->next[i]=NULL;
	root->fa=NULL;
	root->flag=1;
}
void insert(char*str)         //插入单词 
{
	int len=strlen(str);
	Trie*s=root;
	for(int i=0;i<len;i++)
	{
		s->flag=1;              //此前flag=1则表示可以成为某个单词的前缀 
		if(s->next[str[i]-'a'])
			s=s->next[str[i]-'a'];
		else
		{
			Trie*t=new Trie;
			for(int j=0;j<26;j++)
				t->next[j]=NULL;
			t->fa=s;
			t->flag=1;
			s->next[str[i]-'a']=t;
			s=t;
		}
	}
	s->flag=1;
}
int search(char*str)       //查找 
{
	int len=strlen(str);
	Trie*s=root;
	for(int i=0;i<len;i++)
	{
		if(s->next[str[i]-'a']&&s->next[str[i]-'a']->flag)   //存在且flag=1 
			s=s->next[str[i]-'a'];
		else
			return 0;
	}
	return 1;
}
void del(char*str)           //删除节点 
{
	int len=strlen(str);
	Trie*s=root;
	int cnt=0;
	for(int i=0;i<len;i++)         //找出存在的能成为这个单词且比他短的前缀的个数 
	{
		if(s->flag==1)cnt++;
		if(s->next[str[i]-'a'])
			s=s->next[str[i]-'a'];
		else
			return;
	}
	Trie*ss=root;
	int x=0;
	int k;
	for(k=0;k<len;k++)       //使ss最后落在能成为这个单词的前缀的单词的最后一个字母上 
	{
		if(ss->flag==1)x++;
		if(x>=cnt)break;
		if(ss->next[str[k]-'a'])
			ss=ss->next[str[k]-'a'];
	}
	for(;k<len;k++)      //从这个字母开始寻找符合str的节点中子节点只有一个的节点并使其flag=0； 
	{
		if(ss->next[str[k]-'a'])
		{
			int sum=0;
			for(int i=0;i<26;i++)
				if(ss->next[i])sum++;
			if(sum>1)
				ss=ss->next[str[k]-'a'];
			else
			{
				ss->flag=0;
				return;
			}
		}
	}
	ss=ss->fa;
	ss->next[str[k-1]-'a']=NULL;
	return;
}
int main()
{
	init();
	int t;
	cin>>t;
	char op[10],word[35];
	while(t--)
	{
		scanf("%s %s",op,word);
		if(strcmp(op,"insert")==0)
		{
			insert(word);
		}
		else if(strcmp(op,"search")==0)
		{
			if(search(word)==0)cout<<"No"<<endl;
			else cout<<"Yes"<<endl;
		}
		else if(strcmp(op,"delete")==0)
		{
			del(word);
		}
	}
}