codevs 4189 trie树(字典树入门)

题目:http://codevs.cn/problem/4189/

这篇博客讲的超级好:https://www.cnblogs.com/TheRoadToTheGold/p/6290732.html

#include<bits/stdc++.h>
using namespace std;
const int maxx=2000010;
char s[100];
int trie[maxx][26];//trie[rt][x]=tot:rt是上个节点编号,x是字母,tot是下个节点编号
int tot=1;
void insert(char *s)
{
    int rt=1;
    for(int i=0;s[i]!='\0';i++)
    {
        int x=s[i]-'a';
        if(!trie[rt][x])trie[rt][x]=++tot;
        rt=trie[rt][x];
    }
}
bool find(char *s)
{
    int rt=1;
    for(int i=0;s[i]!='\0';i++)
    {
        int x=s[i]-'a';
        if(!trie[rt][x])return false;
        rt=trie[rt][x];
    }
    return true;
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        insert(s);
    }
    int m;
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",s);
        if(find(s))printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}

 

posted @ 2019-06-12 16:10  灰灰烟影  阅读(176)  评论(0编辑  收藏  举报