hdu 1251:统计难题（字典树，经典题）

统计难题

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 16905    Accepted Submission(s): 7273

Problem Description

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

 

Input

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

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

 

Output

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

 

Sample Input

banana

band

bee

absolute

acm

 

ba

b

band

abc

 

Sample Output

2

3

1

0

 

Author

Ignatius.L

 

Recommend

Ignatius.L   |   We have carefully selected several similar problems for you:  1075 1247 1671 1298 1800 

---

 

　　字典树，经典题。

　　今早看了看资料，晚上就会做题了，真的不难，不明白自己为什么放了这么久才看。字典树还是很简单的。

　　我用了2种做法给大家参考，一种是用链表存储，一种是用数组存储。

　　思路：

　　　　字典树的经典应用，给你多个单词，构建字典树，然后给你一个字符串，求以这个字符串为前缀的单词的数量。

　　注意：

　　　　注意如何判断空行，我用了2种方法。

　　　　1、用strlen()计算字符串的长度，如果长度为0，说明为空行，退出输入循环。见代码一

　　　　2、用gets()读入。读入的回车符会自动转换为NULL。所以循环读入，每次检测读入进来的字符串的第一个字符是否为NULL即可。见代码二。

　　代码一（链表）：

#include <iostream>
#include <string.h>
using namespace std;

struct Trie{    //字典树定义
    Trie* next[26];
    int num;    //以当前字符串为前缀的单词的数量
    Trie()    //构造函数
    {
        int i;
        for(i=0;i<26;i++){
            next[i] = NULL;
        }
        num=0;
    }
};
Trie root;
void Insert(char word[])    //将字符串word插入到字典树中
{
    Trie *p = &root;
    int i;
    for(i=0;word[i];i++){    //遍历word的每一个字符
        if(p->next[word[i]-'a']==NULL)    //如果该字符没有对应的节点
            p->next[word[i]-'a'] = new Trie;    //创建一个
        p = p->next[word[i]-'a'];
        p->num++;
    }
}
int Find(char word[])    //返回以字符串word为前缀的单词的数量
{
    Trie *p = &root;
    int i;
    for(i=0;word[i];i++){    //在字典树找到该单词的结尾位置
        if(p->next[word[i]-'a']==NULL)
            return 0;
        p = p->next[word[i]-'a'];
    }
    return p->num;
}

int main()
{
    char word[11];
    while(cin.getline(word,12)){    //输入单词
        if(strlen(word)==0 || word[0]==' ')    //如果读入字符串的长度为0或者是空格，说明读入的是空行
            break;
        Insert(word);
    }
    while(scanf("%s",word)!=EOF){
        printf("%d
",Find(word));    //返回word为前缀的单词的数量
    }
    return 0;
}

 　　代码二（数组）：

#include <iostream>
#include <stdio.h>
using namespace std;

int trie[1000010][26];    //数组形式定义字典树，值存储的是下一个字符的位置
int num[1000010]={0};    //附加值，以某一字符串为前缀的单词的数量
int pos = 1;

void Insert(char word[])    //在字典树中插入某个单词
{
    int i;
    int c = 0;
    for(i=0;word[i];i++){
        int n = word[i]-'a';
        if(trie[c][n]==0)    //如果对应字符还没有值
            trie[c][n] = pos++;
        c = trie[c][n];
        num[c]++;
    }
}
int Find(char word[])    //返回以某个字符串为前缀的单词的数量
{
    int i;
    int c = 0;
    for(i=0;word[i];i++){
        int n = word[i]-'a';
        if(trie[c][n]==0)
            return 0;
        c = trie[c][n];
    }
    return num[c];
}

int main()
{
    char word[11];
    while(gets(word)){
        if(word[0]==NULL)    //空行。gets读入的回车符会自动转换为NULL。
            break;
        Insert(word);
    }
    while(gets(word))
        printf("%d
",Find(word));
    return 0;
}

SUM：

　　以下是2种做法的效率对比：

　　第二行的是第一种做法，第一行的是第二种做法，很明显，数组做法不管是在时间，空间，代码量上都要优于链表做法。

　　但是数组的不太好理解，如何取舍看各位看官的意思吧。

Freecode : www.cnblogs.com/yym2013