杭电acm 5687Problem C（字典树）

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: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串

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;
}