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;
}