HDU 1251-统计难题

题意：省略……（中文题，你看懂的）

思路：就是建一个字典树存放以前出现过的字符串，简单的来说就是相同的前缀可以放在一起，不同的话就开始分支。

查询的时候如果遇到没有走过的节点（t[p].next[x]==-1)，说明没有这个前缀的字符串。

#include<stdio.h>
#include<string.h>
const int maxn=500001;
struct node
{
    int idx;//计数
    int next[26];//存放26个字母
    void init()//没走一个新的节点就初始化
    {
        idx=0;
        memset(next,-1,sizeof(next));
    }
}t[maxn];
int cur=1;
void insert(char *s,int n)
{
    int p=0;
    for(int i=0;i<n;i++)
    {
        int x=s[i]-'a';
        if(t[p].next[x]==-1)//表示以前没有字母到达这个节点
        {
            t[cur].init();//
            t[p].next[x]=cur++;
        }
        p=t[p].next[x];//以此往下分支
        t[p].idx++;//代表这个节点被走过多少遍

    }
}
void reaserch(char *s)
{
    int i=0,p=0;
    while(s[i])
    {
        int x=s[i]-'a';
        if(t[p].next[x]==-1)//如果这个节点没有的话，代表查询的在以前的字典树中没有出现过
        {
            printf("0\n");
            return ;
        }
        p=t[p].next[x];
        i++;
    }
    printf("%d\n",t[p].idx);
}
int main()
{
    char str[101];
    t[0].init();//记得初始化，因为这个找了半天的错
    while(gets(str)!=NULL)
    {
        int len=strlen(str);
        if(len==0) break;
        insert(str,len);
    }
    while(scanf("%s",str)!=EOF)
    {
        reaserch(str);
    }
    return 0;
}