模板 Trie树

模板 Trie树

code：

include

include

using namespace std;

const int wx=20017;

inline int read(){
int sum=0,f=1; char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
return sum*f;
}

struct Trie{
int tr[wx][27];
int cnt;
int flag[wx];

void insert(char *s){
int p=0;
for(int i=0;s[i];i++){
int k=s[i]-'a';
if(!tr[p][k])tr[p][k]=++cnt;
p=tr[p][k];
}
flag[p]=1;
}

bool query(char *s){
int p=0;
for(int i=0;s[i];i++){
int k=s[i]-'a';
if(!tr[p][k])return false;
p=tr[p][k];
}
if(flag[p])return true;
return false;
}
}Trie;

int n;
int m;
char c[wx];

int main(){
n=read();
for(int i=1;i<=n;i++)
scanf("%s",c),Trie.insert(c);
m=read();
for(int i=1;i<=n;i++)
scanf("%s",c),printf("%d\n",Trie.query(c));
return 0;
}