POJ 1251 统计难题（初学字典树）

今天就这题照着别人的代码学了字典树...

这篇论文：http://wenku.baidu.com/view/d2ba836fb84ae45c3b358ca8.html介绍了各种字典树的运用，长了好多姿势，很有启发性

归纳一下字典树的应用：

检索（主要功能）

串排序

在DP中减少无效的状态转移

最长公共前缀问题（LCP）转化成LCA

tire+KMP 构成AC自动机数据结构

Problem Description

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

 

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.

 

Output

对于每个提问,给出以该字符串为前缀的单词的数量.

//93MS 44636K#include<cstdio>#include<iostream>#include<algorithm>#include<cstring>using namespace std;struct trie{    int cnt;    trie *child[26];    trie(){ // 构造函数        cnt=0;        memset(child,0,sizeof(child));    }};trie *root=new trie;  //new比malloc方便trie *current,*newnode;void Insert(char *str){    current=root;    for(int i=0;str[i];i++){        int m=str[i]-'a';        if(current->child[m]!=NULL){            current=current->child[m];            current->cnt++;        }        else {            newnode=new trie;            newnode->cnt++;            current->child[m]=newnode;            current=newnode;        }    }}int Find(char *str){    current=root;    for(int i=0;str[i];i++){        int m=str[i]-'a';        if(current->child[m]){            current=current->child[m];        }        else return 0;    }    return current->cnt;}int main(){    char tmp[20];    while(gets(tmp),strcmp(tmp,"")){        Insert(tmp);    }    while(gets(tmp)!=NULL){        printf("%d\n",Find(tmp));    }    return 0;}