/*
给一个不用指针,用链式前向星写的TRIE,当初只为省空间
*/
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define For(i,a,b) for(register int i=a;i<=b;++i)
#define Rep(i,a,b) for(register int i=a;i>=b;--i)
const int maxx=5000001;
using namespace std;
int be[maxx],ne[maxx],to[maxx],e=0,cnt=0; //cnt表示节点编号,e表示边的编号
void add(int x,int y){
to[++e]=y;
ne[e]=be[x];
be[x]=e;
}
struct node{
char x; //x:当前节点所存的字符
bool vis,end; //vis 是否REPEAT end是否为名字结尾
}trie[maxx];
char name[56];
void insert(char *s){ //插入操作
int pos=0,len=strlen(s),i=0,flag;
For(i,0,len-1){
flag=0;
for(int j=be[pos];j;j=ne[j]){
int go=to[j];
if(trie[go].x==s[i]){ //找到已有相同节点,就把位置赋给pos
flag=1;
pos=go;
break;
}
}
if(!flag) {add(pos,++cnt); pos=cnt; trie[pos].x=s[i];}//没有找到就新建节点,单向边!
if(i==len-1) trie[pos].end=1; //标记结束
}
}
int query(char *s){
int pos=0,len=strlen(s),i=0,flag;
For(i,0,len-1){
flag=0;
for(int j=be[pos];j;j=ne[j]){
int go=to[j];
if(trie[go].x==s[i]){
pos=go;
flag=1;
break;
}
}
if(!flag) return 0; //没找到就返回0
if(i==len-1)
if(trie[pos].end){ //判断是否REPERT
if(!trie[pos].vis) {trie[pos].vis=1; return 1;}
else return 2;
}
else return 0;
}
}
int main(){
#ifndef ONLINE_JUDGE
freopen("input.in", "r", stdin);
freopen("output.out", "w", stdout);
#endif
int n,m,k;
scanf("%d",&n);
For(i,1,n){
scanf("%s",name);
insert(name);
}
scanf("%d",&m);
For(i,1,m){
scanf("%s",name);
k=query(name);
if(k==1) puts("OK");
else if(!k) puts("WRONG");
else puts("REPEAT");
}
return 0;
}
任意门
TRIE(字典树)模板
最新推荐文章于 2019-06-25 18:21:18 发布