HDU 1251 统计难题

简单字典树  这是我初次接触字典树，代码的效率还不是很高，有什么建议，敬请指教

题目：

统计难题

Problem Description

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

Input

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

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

Output

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

Sample Input

banana
band
bee
absolute
acm
ba
b
band
abc

Sample Output

2
3
1
0
下面直接贴代码1  C:

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define max 26
typedef struct t_node  //字典树结点信息
{
    struct t_node * next[max];
    int count;
}t_node;
t_node * root;
void insert(char * str)  //在字典树中新增子树
{
    int i, index, len;
    t_node * current, * newnode;
    len = strlen(str);
    if(len == 0)
        return ;
    current = root;
    for(i = 0; i < len; i ++)
    {
        index = str[i] - 'a';   //寻找对应的子树进行插入
        if(current->next[index] != NULL)  //如果有该结点，继续
        {
            current = current->next[index];
            (current->count) ++;
        }
        else  //如果没有，新增结点
        {
            newnode = (t_node *) calloc(1, sizeof(t_node));
            current->next[index] = newnode;
            current = newnode;
            current->count = 1;
        }
    }
}

int find(char * str) //查看与str对应的前缀的数目
{
    int i, index, len;
    t_node * current = NULL;
    len = strlen(str);
    if(len == 0)
        return 0;
    current = root;
    for(i = 0; i < len; i ++)
    {
        index = str[i] - 'a';
        if(current->next[index] != NULL)  //如果有相应结点，继续
            current = current->next[index];
        else   //如果没有相应结点，返回0
            return 0;
    }
    return current->count;
}

int main()
{
    char str[11];
    root = (t_node *) calloc(1, sizeof(t_node));
    while(gets(str) && strcmp(str, "") != 0)
    {
        insert(str);
    }
    while(scanf("%s", str) != EOF)
    {
        printf("%d\n", find(str));
    }
    return 0;
}

 

代码2  STL：968Ms

#include<iostream>
#include<map>
#include<string>
using namespace std;
int main()
{
    int i, len;
    char str[11];
    map<string, int > m;
    while(gets(str))
    {
        len = strlen(str);
        if(len == 0)
            break;
       
        for(i = len; i >= 1; i --)
        {
            str[i] = '\0';
            m[str] ++;
        }
    }
    while(gets(str))
    {
        cout << m[str] << endl;
    }
    return 0;
}