HDU 1251 - 统计难题（字典树模板题）

题目链接 http://acm.hdu.edu.cn/showproblem.php?pid=1251

【题目描述】
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
对于每个提问,给出以该字符串为前缀的单词的数量.

【样例输入】
banana
band
bee
absolute
acm

ba
b
band
abc

【样例输出】
2
3
1
0

【思路】
模板题，构造一颗Trie树存储所有字符串，并记录前缀数量

//二维数组表示Trie，速度更快，内存更优
#include<bits/stdc++.h>
using namespace std;

const int maxn=1000050;

char s[maxn];
int trie[maxn][26],tot;
int cnt[maxn];

void insert(){
    int node=0;
    for(int i=0;s[i];++i){
        int id=s[i]-'a';
        if(trie[node][id]==0) trie[node][id]=++tot;
        node=trie[node][id];
        ++cnt[node];
    }
}

int query(){
    int node=0;
    for(int i=0;s[i];++i){
        int id=s[i]-'a';
        if(trie[node][id]==0) return 0;
        node=trie[node][id];
    }
    return cnt[node];
}

int main(){
    while(gets(s)){
        if(s[0]=='\0') break;
        insert();
    }
    while(gets(s)){
        int ans=query();
        printf("%d\n",ans);
    }
    return 0;
}

//指针写法，G++会爆内存，C++能过
#include<cstdio>
#include<cstring>
using namespace std;

char s[15];
struct node{
    int cnt;//前缀数量
    node* next[26];
    node(){
        cnt=0;
        memset(next,0,sizeof(next));
    }
}trie;

void insert(){
    node *p=&trie;
    for(int i=0;s[i];++i){
        int id=s[i]-'a';
        if(!p->next[id]) p->next[id]=new node();
        p=p->next[id];
        ++p->cnt;
    }
}

int query(){
    node* p=&trie;
    for(int i=0;s[i];++i){
        int id=s[i]-'a';
        if(!p->next[id]) return 0;
        p=p->next[id];
    }
    return p->cnt;
}

int main(){
    while(gets(s)){
        if(s[0]=='\0') break;
        insert();
    }
    while(gets(s)){
        int ans=query();
        printf("%d\n",ans);
    }
    return 0;
}