HDU2846【字典树】

题意:
百度。

思路:

一个串的插入只能搞出这个串的前缀,然而对于要query的串是子串,所以插入的时候暴力插入所有字母开头的串。。。。
然后还要注意到自己的串本身会叠加字典树中的前缀,要标记掉。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=2e6+10;
int g[N][27],sz,num[N],id[N];
int n,m;

void INS(char *str,int temp)
{
    int p=0;
    int len=strlen(str),index;
    for(int i=0;i<len;i++)
    {
        index=str[i]-'a';
        if(!g[p][index])
        {
            sz++;
            memset(g[sz],0,sizeof(sz));
            g[p][index]=sz;
        }
        p=g[p][index];
        if(id[p]!=temp)
        {
            num[p]++;
            id[p]=temp;
        }
    }
}

int query(char *str)
{
    int p=0;
    int len=strlen(str),index;
    for(int i=0;i<len;i++)
    {
        index=str[i]-'a';
        if(!g[p][index])
            return 0;
        p=g[p][index];
    }
    return num[p];
}

int main()
{
    char s[23];
    scanf("%d",&n);
    memset(g[0],0,sizeof(g[0]));
    sz=0;
    while(n--)
    {
        scanf("%s",s);
        int len=strlen(s);
        for(int i=0;i<len;i++)
        INS(s+i,n);
    }
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",s);
        printf("%d\n",query(s));
    }
    return 0;
}



posted @ 2017-02-16 20:22  see_you_later  阅读(137)  评论(0编辑  收藏  举报