最经,skyzhong得到了一本好厉害的字典，这个字典里整整有n个单词(1<=n<=200000)
现在skyzhong需要在字典里查询以某一段字母开头的单词
如:skyzhong想查询a
那么只要是a开头的单词就可以了
skyzhong只想知道里面有没有这一个单词（因为没有他就不查了）
若有，请输出YES。若没有，请输出NO
第一行一个数n
第二行到第n+1行，一行一个字符串
再下一行一个数m，表示skyzhong想要查询的次数
接着m行，一行一个字符串，表示skyzhong想要查的东西
共m行，若有这字串输出YES，否则输出NO
3
asd
asfdghj
asfd
3
asd
asdghj
asf
YES
NO
YES
字符串只有小写字母，且长度≤8
#include<bits/stdc++.h>
#define N 2000010
using namespace std;
//map<string ,int>W; //map淼
int n,m,len,tot,trie[N][26],sum[N],v[N];
char s[200];
struct node{
int count;
node *next[26];
node(){count=0,memset(next,0,sizeof(next));}
}*rt;
void insert(){
node *r=rt;
char *word=s;
while(*word){
int id=*word-‘a‘;
if(r->next[id]==NULL) r->next[id]=new node;//r->next[id]=build();
r=r->next[id];
r->count++;
word++;
}
}
int search(){
node *r=rt;
char *word=s;
while(*word){
int id=*word-‘a‘;
r=r->next[id];
if(r==NULL) return 0;
word++;
}return r->count;
}
int main()
{
rt=new node;
cin>>n;
for(int i=1;i<=n;i++){
cin>>s;
insert();
}cin>>m;
for(int i=1;i<=m;i++){
cin>>s;
if(search()!=0){
cout<<"YES\n";
}else cout<<"NO\n";
}return 0;
}
2.数组模拟板
#include<bits/stdc++.h>
#define N 2000010
using namespace std;
//map<string ,int>W; //map淼
int n,m,len,tot,trie[N][26],rt,sum[N],v[N];
char s[200];
void insert(){
len=strlen(s);
rt=0;
for(int i=0;s[i];i++){
int id=s[i]-‘a‘;
if(!trie[rt][id]) trie[rt][id]=++tot;
sum[trie[rt][id]]++;
rt=trie[rt][id];
}
}
int search(){
len=strlen(s);
rt=0;
for(int i=0;s[i];i++){
int id=s[i]-‘a‘;
if(!trie[rt][id]) return 0;
rt=trie[rt][id];
}return sum[rt];
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++){
cin>>s;
insert();
}cin>>m;
for(int i=1;i<=m;i++){
cin>>s;
if(search()!=0){
cout<<"YES\n";
}else cout<<"NO\n";
}return 0;
}
3.Map水题版
#include<bits/stdc++.h>
using namespace std;
map<string ,int>W;
int n,m,len;
char s[200];
int main()
{
cin>>n;
for(int i=1;i<=n;i++){
cin>>s;
len=strlen(s);
for(int i=len;i>=1;i--){
s[i]=‘\0‘;
W[s]++;
}
}cin>>m;
for(int i=1;i<=m;i++){
cin>>s;
if(W[s]!=0){
cout<<"YES\n";
}else cout<<"NO\n";
}return 0;
}
Trie树初探，参考博客：TRTTG
总结操作：插入+查找
原文：https://www.cnblogs.com/song-/p/9191981.html