hdoj1251-字典树入门水题-优快云博客

本文链接：https://blog.youkuaiyun.com/guogai13/article/details/79872245

解决一个统计问题，通过构建特殊的数据结构来高效地统计给定单词列表中以特定字符串为前缀的单词数量。

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

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

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

注意:本题只有一组测试数据,处理到文件结束.
Output 对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

Sample Output

#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
char s[1000];
int k,ans;
struct Tree{
    int A[30],v;
    ins(){
       memset(A,0,sizeof(A));
       v=0;
    }
}T[500010];
void build(int f,int l){
    int i,n=strlen(s),v=s[l]-'a';
    if(l==n) return;
    if(T[f].A[v]==0){
        k++;
        T[f].A[v]=k;
    }
    T[T[f].A[v]].v++;
    build(T[f].A[v],l+1);
}
int q(int f,int l){
    int i,n=strlen(s),v=s[l]-'a';
    if(l==n) return T[f].v;
    if(T[f].A[v]==0) return 0;
    return q(T[f].A[v],l+1);
}
int main(){
    k=0;
    int i;
    for(i=0;i<500000;i++) T[i].ins();
    while(gets(s)){
        if(strlen(s)==0) break;
        build(0,0);
    }
    while(gets(s)!=NULL){
        ans=q(0,0);
        printf("%d\n",ans);
    }
    return 0;
}