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");
	}
}