Hdu.1251.统计难题（字典树）

最新推荐文章于 2021-11-15 22:42:07 发布

MiaoPlus

最新推荐文章于 2021-11-15 22:42:07 发布

阅读量186

点赞数

CC 4.0 BY-SA版权

分类专栏： =========基本数据结构========== 文章标签： Hdu.1251.统计难题字典树(Trie)

本文链接：https://blog.youkuaiyun.com/XxxxxM1/article/details/80205978

=========基本数据结构========== 专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种使用字典树解决统计单词前缀数量的方法。文章提供了两种解决方案，一种是利用标准模板库中的映射类型，另一种是通过构建字典树进行高效的查询。这两种方法都实现了快速统计单词集中以特定字符串作为前缀的单词数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

B - 统计难题

Time Limit: 2000 MS Memory Limit: 65535 KB

64-bit integer IO format: %I64d , %I64u Java class name: Main

[Submit] [Status]

Description

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

Input

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

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

Output

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

Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

考点：（Trie）字典树

方法一：思维

# include <stdio.h>
# include <string>
# include <map>
#include<iostream>
using namespace std;
int main()
{
    char a;
    string  x;
    map<string,int>q;
    while(true){
        scanf("%c",&a);
        if(a=='\n'){
            scanf("%c",&a);
            x="";
        }
        if(a=='\n')  break;
        x+=a;
        q[x]+=1;
    }
    while(cin>>x){
        printf("%d\n",q[x]);}
    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\n",Find(word));
    return 0;
}