2016"百度之星" - 资格赛（Astar Round1） Problem C

字典树。

插入的时候update一下节点出现的次数。

delete的时候，先把前缀之后的全删了。然后看前缀最后一个节点出现了几次，然后前缀上每个节点的次数都减去这个次数。

前缀从上到下再检查一遍，如果出现了次数为0的节点，也删去。

#include <stdio.h>
#include <math.h>
#include<cstring>
#include<cmath>
#include<map>
#include<string>
#include<algorithm>
using namespace std;

struct Node
{
    int val;
    int f[];
} node[];

int root,n;
char op[], s[];
int tot;

int newnode()
{
    ++tot;
    node[tot].val=;
    for(int i=; i<; i++) node[tot].f[i]=-;
    return tot;
}

void Insert()
{
    int p=root;
    for(int i=; s[i]; i++)
    {
        if(node[p].f[s[i]-'a']==-) node[p].f[s[i]-'a']=newnode();
        p=node[p].f[s[i]-'a'];
        node[p].val++;
    }
}

void Delete()
{
    int p=;
    bool Find=;
    for(int i=; s[i]; i++)
    {
        if(node[p].f[s[i]-'a']==-)
        {
            Find=;
            break;
        }
        p=node[p].f[s[i]-'a'];
    }
    if(Find==) return;

    for(int i=; i<; i++) node[p].f[i]=-;

    int Val=node[p].val;

    p=root;
    for(int i=; s[i]; i++)
    {
        p=node[p].f[s[i]-'a'];
        node[p].val=node[p].val-Val;
    }

    p=;
    for(int i=; s[i]; i++)
    {
        int tmp=node[p].f[s[i]-'a'];
        if(node[tmp].val==)
        {
            node[p].f[s[i]-'a']=-;
            break;
        }
        p=node[p].f[s[i]-'a'];
    }
}

void Search()
{
    int p=root; bool flag=;
    for(int i=; s[i]; i++)
    {
        if(node[p].f[s[i]-'a']==-)
        {
            flag=;
            break;
        }
        p=node[p].f[s[i]-'a'];
    }

    if(flag) printf("No\n");
    else printf("Yes\n");
}

int main()
{
    while(~scanf("%d",&n))
    {
        tot=; root=;
        for(int i=; i<; i++) node[root].f[i]=-;
        for(int i=; i<=n; i++)
        {
            scanf("%s%s",op,s);
            if(strcmp(op,"insert")==) Insert();
            else if(strcmp(op,"delete")==) Delete();
            else if(strcmp(op,"search")==) Search();
        }
    }
    return ;
}