hdu1251解题报告

题目大意：给一个单词表，再给一个前缀表，问每个前缀在单词表中出现的次数？

解题思路：前缀树，Trie，字典树，每个结点记录插入单词时经过这个结点的次数。

注意点：没说单词的最长的长度，使用静态结构的话开5000005个结点应该够，动态结构则可以忽略这个问题。

ac代码如下：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=5000005;
const int sigma_size=26;
int ch[maxn][sigma_size];
int val[maxn];
int sz;

int idx(char c){ return c-'a'; }

void Insert(char *s)
{
int u=0;
int n=strlen(s);
for(int i=0;i<n;i++)
{
int c=idx(s[i]);
if(!ch[u][c])
{
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=1;
ch[u][c]=sz++;
}
else
{
val[ch[u][c]]++;
}
u=ch[u][c];
}
}

int query(char *s)
{
int u=0;
int n=strlen(s);
for(int i=0;i<n;i++)
{
int c=idx(s[i]);
if(!ch[u][c])
{
return 0;
}
if(i==n-1)
{
return val[ch[u][c]];
}
u=ch[u][c];
}
}

int main()
{
//freopen("1.txt","r",stdin);
char String[30];
sz=1;
memset(ch[0],0,sizeof(ch[0]));
while(gets(String)&&String[0]!='\0') Insert(String);
while(gets(String)) printf("%d\n",query(String));
}