首先讲一下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;
}