戏说Trie字典树算法+Luogu2580 于是他错误的点名开始了

首先讲一下Trie这个东西(其实很简单)

在计算机科学中,trie,又称前缀树或字典树,是一种有序树,用于保存关联数组,其中的键通常是字符串。与二叉查找树不同,键不是直接保存在节点中,而是由节点在树中的位置决定。一个节点的所有子孙都有相同的前缀,也就是这个节点对应的字符串,而根节点对应空字符串。

以上摘自百度百科
这就像英文字典一样,每个字母按照字典序排
举个例子,比如我们要插入几个字符串:

a
aabc
abdc
bac
bacc

插入好的trie长成这样:
这里写图片描述
其中绿色的节点为某个字符串尾
这样是不是就能很快地存储字符串和比较了?
于是我们就能在这棵树上跑很多的操作了

1.插入

很好理解,只要记录一下每个点对应的子节点编号是哪个就行了

inline void insert(char c[]){
    ll now=0,l=strlen(c);
    for(int i=1;i<=l;i++){
        ll t=c[i]-'a';//我这里看作字符串中只有小写字母
        if(!p[now][t])p[now][t]=++np;//于是这个数组直接记录子节点字母对应的位置即可
        now=p[now][t];//更新节点编号
    }
    b[now]=1;//最后打个字符串尾部标记(即图中绿色节点)
}

2.判断及去重

也很好理解,一个一个字符找下去判断即可

inline int check(char c[]){
    ll now=0,l=strlen(c);
    for(int i=1;i<=l;i++){
        ll t=c[i]-'a';//我这里看作字符串中只有小写字母
        if(!p[now][t])return 2;//发现Trie中并没有这个字符串
        now=p[now][t];//更新节点编号
    }
    if(b[now]==1){b[now]=2;return 0;}//表示这个字符串在目前时刻只出现过一次
    return 1;//表示在此之前这个字符串已经出现过了(即b[now]已经是2了)
}

其实其他还有不少操作,比如说用来跑AC自动机(不过我还不会。。。
这里先介绍最简单的操作,其它的以后再说。。。


例题:Luogu2580 于是他错误的点名开始了
https://www.luogu.org/problem/show?pid=2580
把上面的两个操作拼起来就好了,模板题

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
int np=0,p[800001][26]={0};
int b[1000001]={0};
char c[1000001];
inline void insert(){
    ll now=0,l=strlen(c+1);
    for(int i=1;i<=l;i++){
        ll t=c[i]-'a';
        if(!p[now][t])p[now][t]=++np;
        now=p[now][t];
    }
    b[now]=1;
}
inline int check(){
    ll now=0,l=strlen(c+1);
    for(int i=1;i<=l;i++){
        ll t=c[i]-'a';
        if(!p[now][t])return 2;
        now=p[now][t];
    }
    if(b[now]==1){b[now]=2;return 0;}
    return 1;
}
int main()
{
    int n;scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",c+1);
        insert();
    }
    int m;scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s",c+1);
        int k=check();
        if(k==2)printf("WRONG\n");
        else if(k==1)printf("REPEAT\n");
        else printf("OK\n");
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值