HDU 5687 Problem C 【字典树删除】

传..传送：http://acm.hdu.edu.cn/showproblem.php?pid=5687

Problem C

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

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）

 

解题思路：

输入查询都是字典树基本操作，这里的字典树的删除是一种标记删除而不是物理删除（非最后节点）。

但是这里的标记删除要注意的是不能直接取消标记或标记为 0 ，而是要根据要删除的字符串在字典中出现的次数，然后根据这个次数去更新路径上的点的标记。如果是删除字符的最后节点则进行物理删除。

例如说：字典里是 zzaizjja，zzaizjj，要删除 zzaizjja，如果单纯把路径 zzaizjja 上的节点都更新为0，则它的子串也 out 了，所以正确的更新是：路径上的点减去zzaizjja出现的次数。

AC Code：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#define INF 0x3f3f3f3f
#define LL long long int
#define Bits 26
using namespace std;
const int MAXN = 33;
int N;
struct Trie
{
    Trie *next[Bits];
    int flag;
    inline void init(){
        this->flag = 1;
        for(int i = 0; i < Bits; i++)
            this->next[i] = NULL;
    }
};
Trie *Root = (Trie *)malloc(sizeof(Trie));
inline void DelTrie(Trie *t)
{
    if(t == NULL) return;
    for(int i = 0; i < Bits; i++){
        if(t->next[i]!=NULL) DelTrie(t->next[i]);
    }free(t);
    return;
}

void Create_Trie(char *str, bool isDel)
{
    int len = strlen(str);
    Trie *p = Root, *temp;
    int Delnum = 0;
    for(int i = 0; i < len; i++){
        int index = str[i]-'a';
        if(!isDel){
            if(p->next[index] == NULL){
                temp = (Trie *)malloc(sizeof(Trie));
                temp->init();
                p->next[index] = temp;
                p = p->next[index];
            }
            else{p->next[index]->flag+=1; p=p->next[index];}
        }
        else{
            if(p->next[index] != NULL){
                Delnum = p->next[index]->flag;
                p = p->next[index];
            }else return;
        }
    }
    if(isDel){
        p = Root;
        for(int i = 0; i < len; i++){
            int index = str[i]-'a';
            if(p->next[index] == NULL) return;
            if(i==len-1){
                DelTrie(p->next[index]);
                p->next[index] = NULL;
                return;
            }
            p->next[index]->flag-=Delnum;
            p = p->next[index];
        }
    }
}


bool Find_Trie(char *str)
{
    int len = strlen(str);
    Trie *p = Root;
    for(int i = 0; i < len; i++){
        int index = str[i]-'a';
        if(p->next[index] != NULL && p->next[index]->flag > 0)
            p=p->next[index];
        else return false;
    }return true;
}

int main()
{
    Root->init();
    scanf("%d", &N);
    char str[MAXN], command[10];
    while(N--){
        scanf("%s", command);
        if(strcmp(command, "insert")==0){
            scanf("%s", str);
            Create_Trie(str, false);
        }
        else if(strcmp(command, "delete")==0){
            //puts("zjj");
            scanf("%s", str);
            Create_Trie(str, true);
        }
        else{
            scanf("%s", str);
            if(Find_Trie(str)) puts("Yes");
            else puts("No");
        }
    }DelTrie(Root);
    return 0;
}