HDU 5687 (字典树)

Problem C

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

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 <cstring>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
#define maxn 3111111

struct node {
    int num;
    int next[26], fa;
}tree[maxn];

char op[33], a[33];
int root, cnt;

void New (int c, int fa) {
    tree[c].num = 0;
    tree[c].fa = fa;
    memset (tree[c].next, -1, sizeof tree[c].next);
    return ;
}

void Insert () {
    int l = strlen (a);
    int p = root;
    for (int i = 0; i < l; i++) {
        int id = a[i]-'a';
        if (tree[p].next[id] == -1) {
            tree[p].next[id] = ++cnt;
            New (cnt, p);
        }
        p = tree[p].next[id];
        tree[p].num++;
    }
    return ;
}

int Search () {
    int l = strlen (a);
    int p = root;
    for (int i = 0; i < l; i++) {
        int id = a[i]-'a';
        if (tree[p].next[id] == -1) {
            return 0;
        }
        p = tree[p].next[id];
    }
    return tree[p].num;
}

void Delete () {
    int l = strlen (a);
    int p = root;
    for (int i = 0; i < l; i++) {
        int id = a[i]-'a';
        if (tree[p].next[id] == -1) {
            return ;
        }
        p = tree[p].next[id];
    }
    int ans = tree[p].num;
    while (tree[p].fa != -1) {
        tree[p].num -= ans;
        p = tree[p].fa;
    }

    p = root;
    for (int i = 0; i < l; i++) {
        int id = a[i]-'a';
        if (tree[tree[p].next[id]].num == 0) {
            tree[p].next[id] = -1;
            return ;
        }
        p = tree[p].next[id];
    }
    return ;
}

int main () {
    //freopen ("in.txt", "r", stdin);
    int n;
    scanf ("%d", &n);
    root = cnt = 0;
    New (root, -1);
    while (n--) {
        scanf ("%s%s", op, a);
        if (op[0] == 'i') {
            Insert ();
        }
        else if (op[0] == 'd') {
            Delete ();
        }
        else if (op[0] == 's') {
            int ans = Search ();
            printf ("%s\n", ans ? "Yes" : "No");
        }
    }
    return 0;
}