hdu 1251 字典树模板题 ---多串 查找单词出现次数

这道题题目里没有给定数据范围 我开了2005  疯狂的WA

然后开了50000， A掉  我以为自己模板理解错  然后一天没吃饭，饿得胃疼还是想着把这题A掉再去吃，谁知居然是这种问题，，，呵呵~~~

不过记录下这道题学到的方法吧：

    for(rt = 0; *s; rt = nxt, ++s)    {        nxt=tree[rt][*s-tb];        if(!nxt)        {            nxt=tree[rt][*s-tb]=top;            memset(tree[top],0,sizeof(tree[top]));            top++;        }        tree[nxt][tk]++;//////////    }

//////////////处，如果改为tree[rt][tk]表示某个节点有几个孩子，tree[nxt][tk]++经过当前字母的单词个数

#include <cstdio>#include <cstring>#include <algorithm>#include <iostream>using namespace std;const int tk=26;const int tb='a';const int MAXN= 500000;int top,tree[MAXN][tk+2];void init(){    top=1;    memset(tree[0],0,sizeof(tree[0]));}void Insert(char *s, int Rank=1){    int rt,nxt;    for(rt = 0; *s; rt = nxt, ++s)    {        nxt=tree[rt][*s-tb];        if(!nxt)        {            nxt=tree[rt][*s-tb]=top;            memset(tree[top],0,sizeof(tree[top]));            top++;        }        tree[nxt][tk]++;    }}int sea(char *s){    int rt;    for(rt=0;rt=tree[rt][*s-tb];)    {        if(*(++s)==0)return tree[rt][tk];    }    return 0;}char str[150],pat[150];int main(){   //freopen("hdu1251.txt","r",stdin);    char cc;    int ans=0;    init();    while(gets(str)&&str[0])    {        Insert(str);    }    while(~scanf("%s",pat))    {        //cout << pat << endl;        printf("%d\n",sea(pat));    }    return 0;}