Trie树初学

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
const int root=0;
int tot;
struct node
{
    int cnt;
    int next[26];
    void newnode()
    {
        cnt=0;
        for(int i=0;i<26;i++)
        {
            next[i]=-1;
        }
    }
}t[1000005];
void clear()
{
    tot=0;
    t[root].newnode();
}
void insert(char *str)
{
    int p=root;
    int len=strlen(str);
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(t[p].next[id]==-1)
        {
            t[++tot].newnode();
            t[p].next[id]=tot;
        }
        p=t[p].next[id];
        t[p].cnt++;
    }
}
int query(char *str)
{
    int p=root;
    int len=strlen(str);
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(t[p].next[id]==-1)
        {
            return 0;
        }
        p=t[p].next[id];
    }
    return t[p].cnt;
}
int main()
{
    int n,m;
    char s[15];
    clear();
    scanf("%d",&n);
    while(n--)
    {
        scanf("%s",s);
        insert(s);
    }
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",s);
        printf("%d\n",query(s));
    }
    return 0;
}
posted @ 2015-05-02 01:29  BYYB_0506  阅读(204)  评论(0编辑  收藏  举报