trie字典树-初级入门
题目大意: 给定N个字符串,随后有M次询问,每次询问一串字符串是否在原先给出的N个字符串里,你的回答需要分情况讨论: 如果这个字符串在原先给定的字符串中出现过,并且这是这个字符串第一次被询问,则输出“OK”,如果之前已经询问过该字符串,则输出“REPEAT”,若没有出现过,则输出“WRONG”。
分析
上面的标题已经透露了,这是一道字典树导弹裸题。因此不再做过多的赘述。
(这里由于理解问题,我给出的trie树并不是递归的)
(个人认为已经非常简洁了)
注意在某个字符串第一次被询问后打上标记!!!
#include<bits/stdc++.h>
using namespace std;
int n,m;
char a[100000];
int son[500000][30],cnt;
int tot[500000];
void insert(char *s){
int len = strlen(s+1),cur=1;
for(int i = 1;i <= len; ++i){
if(son[cur][s[i]-'a'])cur = son[cur][s[i]-'a'];
else son[cur][s[i]-'a']= ++cnt,cur=cnt;
}
tot[cur]++;
}
void find(char *s){
int len = strlen(s + 1),cur = 1;
for(int i = 1;i <= len ; ++i){
cur = son[cur][s[i]-'a'];
}
if(tot[cur] == 1){
puts("OK");
tot[cur] = -1;
}else if (tot[cur] == -1){
puts("REPEAT");
} else {
puts("WRONG");
}
}
int main(){
cin>>n;
for(int i=1;i<=n;i++){
scanf("%s",a+1);
insert(a);
}
cin>>m;
for(int i=1;i<=m;i++){
scanf("%s",a+1);
find(a);
}
}