HDU - 1251 统计难题（字典树模板题）

关注

原创最新推荐文章于 2026-03-03 00:35:03 发布 401 阅读 AI 写同款· GEO 优化 ›

AI 赋能测试全生命周期：多 Agent 协同的自动化提效方案本文提出了一种基于Coze平台的多Agent协同测试解决方案，旨在解决软件测试中PRD解析不清晰、测试点覆盖不全、用例编写效率低、代码生成耗时等痛点。通过PRD优化、测试点生成、测试用例生成和单元测试框架生成四个Agent的协同工作，结合Cursor工具实现从需求到代码的全链路自动化。以电商平台登录模块为例，展示了从模糊PRD到标准化测试用例和可执行Java单元测试代码的完整转化流程。该方案通过AI赋能显著提升测试效率，减少人工干预，实现测试覆盖率≥95%，为快速迭代的软件研发提供了高效的测试自动化支持。阅读详情

统计难题

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

Problem Description

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

Input

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

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

Output

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

Sample Input

  
   banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

Author

Ignatius.L

Recommend

Ignatius.L

Statistic | Submit | Discuss | Note

#include <bits/stdc++.h>
using namespace std;

const int N = 1e6 + 10;
int tot;
char s[1001];
int nex[N][26], cnt[N];

void Init(){
    tot = 0;
    memset(nex[0], -1, sizeof nex[0]);
    cnt[0] = 0;
}

void Insert(char a[]){
    int l = strlen(a);
    int p = 0;
    for(int i = 0; i < l; i++){
        int id = a[i]-'a';
        if(nex[p][id] == -1){
            cnt[++tot] = 0;
            nex[p][id] = tot;
            memset(nex[tot], -1, sizeof nex[tot]);
        }
        p = nex[p][id];
        cnt[p]++;
    }
}

int Query(char a[]){
    int l = strlen(a);
    int p = 0;
    for(int i = 0; i < l; i++){
        int id = a[i]-'a';
        if(nex[p][id] == -1) return 0;
        p = nex[p][id];
    }
    return cnt[p];
}

int main(){
    Init();
    while(gets(s) && strcmp(s, "") != 0){
        Insert(s);
    }
    while(scanf("%s", s) == 1){
        printf("%d\n", Query(s));
    }
}