Problem C
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 131072/131072 K (Java/Others)
Total Submission(s): 2669 Accepted Submission(s): 736
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
Source
2016"百度之星" - 资格赛(Astar Round1)
问题链接:HDU5687 Problem C
问题描述:(略)
问题分析:
这是一个字典树的问题。删除一个单词的做法比较特殊一些,需要注意。
程序说明:
这里采用静态存储分配,删除单词是存储并没有回收,会有一定的浪费。程序倒是AC了。
参考链接:(略)
题记:(略)
AC的C++语言程序如下:
/* HDU5687 Problem C */
#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
const int N = 100000;
const int LEN = 30;
const int SIZE = 26;
const char SCHAR = 'a';
struct Trie {
int acnt; // access count
int childs[SIZE];
void init()
{
acnt = 1;
memset(childs, 0, sizeof(childs));
}
} trie[N * LEN];
int ncnt; // Trie Node count
char cmd[8], s[LEN + 1]; // Input
void insert(char s[])
{
int p = 0;
for(int i = 0; s[i]; i++) {
int k = s[i] - SCHAR;
int child = trie[p].childs[k];
if(child) {
trie[child].acnt++;
p = child;
} else {
trie[++ncnt].init();
trie[p].childs[k] = ncnt;
p = ncnt;
}
}
}
int query(char s[])
{
int p = 0;
for (int i = 0; s[i]; i++) {
int k = s[i] - SCHAR;
int child = trie[p].childs[k];
if (child == 0)
return 0;
else
p = child;
}
return trie[p].acnt;
}
void remove(char s[])
{
int p = 0, last = 0, k;
for (int i = 0; s[i]; i++) {
k = s[i] - SCHAR;
int child = trie[p].childs[k];
if (child == 0)
return;
else {
last = p;
p = child;
}
}
trie[last].childs[k] = 0;
int cnt = trie[p].acnt;
p = 0;
for (int i = 0; s[i + 1]; i++) {
k = s[i] - SCHAR;
p = trie[p].childs[k];
trie[p].acnt -= cnt;
}
}
int main()
{
int n;
scanf("%d", &n);
while(n--) {
scanf("%s%s", cmd, s);
if(cmd[0] == 'i')
insert(s);
else if(cmd[0] == 's')
printf("%s\n", query(s) ? "Yes" : "No");
else if(cmd[0] == 'd')
remove(s);
}
return 0;
}