Problem C
Time Limit : 2000/1000ms (Java/Other) Memory Limit : 131072/131072K (Java/Other)
Total Submission(s) : 1 Accepted Submission(s) : 1
Font: Times New Roman | Verdana | Georgia
Font Size: ← →
Problem Description
度熊手上有一本神奇的字典,你可以在它里面做如下三个操作:
1、insert : 往神奇字典中插入一个单词
2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词
3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串
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
Source
2016"百度之星" - 资格赛(Astar Round1)
想法:字典树模板
代码:
#include<stdio.h>
#include<string.h>
using namespace std;
typedef long long ll;
const int N = 26;
struct Trie
{
int num;
Trie *nx[N];
};
void init(Trie *t)
{
for (int i = 0;i < N;++i)
{
t->nx[i] = NULL;
}
t->num = 0;
}
Trie *root;
void Insert(char w[])
{
Trie *p = root;
int len = strlen(w);
for (int i = 0;i < len;++i)
{
int id = w[i] - 'a';
if (p->nx[id] == NULL)
{
Trie *t = new Trie;
init(t);
p->nx[id] = t;
}
p = p->nx[id];
p->num++;
}
}
int Find(char w[])
{
Trie *p = root;
int len = strlen(w);
for (int i = 0;i < len;++i)
{
int id = w[i] - 'a';
if (p->nx[id] == NULL) return 0;
p = p->nx[id];
}
return p->num;
}
void Delete(char w[],int num)
{
Trie *p = root;
int len = strlen(w);
for (int i = 0;i < len;++i)
{
int id = w[i] - 'a';
if (p->nx[id] == NULL) return;
p->num -= num;//delete
p = p->nx[id];
}
p->num = 0;
for (int i = 0;i < N;++i) p->nx[i] = NULL;
}
int main()
{
char op[110],word[330];
int T;
while (scanf("%d",&T) == 1)
{
root = new Trie;
init(root);
while (T--)
{
scanf("%s %s",op,word);
if (!strcmp(op,"insert"))
{
Insert(word);
}
else if (!strcmp(op,"search"))
{
if (Find(word)) puts("Yes");
else puts("No");
}
else if (!strcmp(op,"delete"))
{
int num = Find(word);
if (num) Delete(word,num);
}
}
}
return 0;
}