字典树hdu5687------8.4 xiaolver正式开始学习的第一天

给自己的话——之前落下的你找时间补吧......哈哈哈哈哈嗝.

先放一个字典树的题.邻接链表确实比邻接矩阵更快更不占内存但是对于我这菜狗子来说理解真是很刺激的一个过程。

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=5687

度熊手上有一本神奇的字典,你可以在它里面做如下三个操作: 

  1、insert : 往神奇字典中插入一个单词 

  2、delete: 在神奇字典中删除所有前缀等于给定字符串的单词 

  3、search: 查询是否在神奇字典中有一个字符串的前缀等于给定的字符串

Input

这里仅有一组测试数据。第一行输入一个正整数N(1≤N≤100000)N(1≤N≤100000),代表度熊对于字典的操作次数,接下来NN行,每行包含两个字符串,中间中用空格隔开。第一个字符串代表了相关的操作(包括: 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

显然此题应用字典树来解决.

首先是Insert

void Insert(char str[])
{
    int c=0;
    for(int i=0;str[i];i++){
        int n=str[i]-'a';    
        if(trie[c][n]==0)               如果从c到n在之前未出现,用pos存从c到n的索引(可以理解为将c刚出现的n直接存成数字信息,即吧一段字符和数字对应起来,num【c】就是这段字符段数量).
            trie[c][n]=pos++;
        c=trie[c][n];
        num[c]++;
    }
}

然后是查询函数

int Find(char word[])
{
    int i;
    int c=0;
    for(i=0;word[i];i++)
    {
        int n=word[i]-'a';
        if(trie[c][n]==0)  //查询这个字符串,如果匹配到某一个字符无法继续匹配下去就返回0 
            return 0;
        c=trie[c][n];
    }return num[c];
}

最后是删除函数

void deletee(char word[])
{    int c;
    int len=strlen(word);int fuck=0;
    xh(i,0,len-1){
    int id=word[i]-'a';
    if(trie[fuck][id]==0)return;//如果要删除的前缀词不存在就结束
    fuck=trie[fuck][id];
    }
    int ans=num[fuck];
    for(int i=0;i<=25;i++)
    {
        c=trie[fuck][i];trie[fuck][i]=0; //前缀匹配结束后能向下匹配的所有字符串清空
        num[c]=0;
    }c=0;
    xh(i,0,len-1)// 前缀前一个词开始所有数目减去词数
    {
        int id=word[i]-'a';
        c=trie[c][id];
        num[c]-=ans;
    }
}

上完整代码

#include <iostream>
#include <string>
#include <cstring>
#include <stdio.h>
#define sd(qwe) scanf("%d",&qwe)
#define ss(asd) scanf("%s",asd)
#define xh(ert,rty,tyu) for(int ert=rty;ert<=tyu;ert++)
#define fxh(u,i,o) for(int u=i;u>=o;u--)

using namespace std;

int trie[1000010][26],num[1000010]={0},pos=1;
void Insert(char str[])
{
    int c=0;
    for(int i=0;str[i];i++){
        int n=str[i]-'a';
        if(trie[c][n]==0)
            trie[c][n]=pos++;
        c=trie[c][n];
        num[c]++;
    }
}
int Find(char word[])
{
    int i;
    int c=0;
    for(i=0;word[i];i++)
    {
        int n=word[i]-'a';
        if(trie[c][n]==0)
            return 0;
        c=trie[c][n];
    }return num[c];
}
void deletee(char word[])
{    int c;
    int len=strlen(word);int fuck=0;
    xh(i,0,len-1){
    int id=word[i]-'a';
    if(trie[fuck][id]==0)return;
    fuck=trie[fuck][id];
    }
    int ans=num[fuck];
    for(int i=0;i<=25;i++)
    {
        c=trie[fuck][i];trie[fuck][i]=0;
        num[c]=0;
    }c=0;
    xh(i,0,len-1)
    {
        int id=word[i]-'a';
        c=trie[c][id];
        num[c]-=ans;
    }


}
int main()
{
    char caozuo[11],word[35];
    int k;sd(k);
    xh(i,1,k)
    {
        ss(caozuo);ss(word);
        if(caozuo[0]=='i') {Insert(word);continue;}
        if(caozuo[0]=='d') {deletee(word);continue;}
        if(caozuo[0]=='s') if(Find(word)>0){cout<<"Yes"<<endl;continue;}cout<<"No"<<endl;
        }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_35557621/article/details/81414536