HDU5687 Problem C【字典树】

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