Codevs4189 字典

#include<iostream>#include<cstring>#include<cstdio>#include<algorithm>using namespace std;int trie[1600001][26],len,root,tot,sum[1600001]; char s[11];//这里的s设为全局变量，插入和查询都是用的这个s void insert(){    len=strlen(s);    root=0;    for(int i=0;i<len;i++)    {        int id=s[i]-'a';        if(!trie[root][id]) trie[root][id]=++tot;  //如果root子节点不含id所表示的字母，那么给trie[root][id]编号         //sum[trie[root][id]]++;//sum表示以trie[root][id]为前缀的单词出现的的次数         root=trie[root][id];//更改根节点，继续向下     }}bool search(){    root=0;    len=strlen(s);//此时的s为要查询的前缀     for(int i=0;i<len;i++)    {        int id=s[i]-'a';        if(!trie[root][id]) return 0;        root=trie[root][id];    }    return 1;}int main(){int n,m;scanf("%d",&n);while(n--){cin>>s;insert();}scanf("%d",&m);while(m--){cin>>s;if(search()) puts("YES\n");else puts("NO\n");}}