HDU 1251(字典树改造)

十分无耻的题目 开了300W的内存才过,这个测试数据也太变态了吧~~~

 

把原先用以比较结束的布尔值进行了替换,加入num值表示是进行了多少次前缀匹配,只要通过查找,返回匹配数就行了~~~上次的模版的某些部分有问题,重新进行了修正...

 

 

#include<iostream>
#include <queue>
using namespace std;
struct node
{
    long num;
    int next[26];
    void init(){memset(next,-1,sizeof(next));num=0;}
}s[3000000];

int p;
char a[100];

inline void preprocess(){s[p=0].init();}

inline void insert()
{
    int index=0;
    ++s[index].num;
    for(int i=0;a[i]!='\0';i++)
    {
        int x=a[i]-'a';
        if(s[index].next[x]==-1)
        {
            s[++p].init();
            s[index].next[x]=p;
        }
        index=s[index].next[x];
        s[index].num++;
    }
}

inline long ct()
{
    long index=0;
    long i;

    for (i=0;a[i]!='\0';++i)
    {
        long x=a[i]-'a';
        if (s[index].next[x]==-1) return 0;
        index=s[index].next[x];
    }

    return s[index].num;
}


int main()
{
    preprocess();

    while (gets(a))
    {

        if(a[0]=='\0')
        {
            break;
        }    

        insert();
    }

    while(gets(a))
    {
        printf("%ld\n",ct());
    }

    return 0;
}