HDU1251统计难题(字典树)

Posted 2022-12-01 沐茈静

题目：

Problem Description Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).   Input 输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串. 注意:本题只有一组测试数据,处理到文件结束.   Output 对于每个提问,给出以该字符串为前缀的单词的数量.   Sample Input banana band bee absolute acm ba b band abc   Sample Output 2 3 1 0

题解：用字典树存储单词，统计每种可能前缀的单词数；

树的每个结点可以有26个子树；对应26个小写字母；（拓展：若单词中包括大写字母则是52个子树，若还有数字，则是62个子树，根据具体题目而定）；

每个节点的数据部分存放统计从根到达此节点的共同前缀的个数；

错误分析：1.输入格式用gets(),一次输入一行；

                 2.q的子节点没有创建并赋空；

#include<stdio.h>
#include<cstring>
#include<stdlib.h>
#include<iostream>
using namespace std;
#pragma comment(linker,"/STACK:102400000,102400000")
#define MAX   26
char a[15];
struct Tritree   //存储结构

     int count;     //标记共同前缀的单词数
     Tritree *next[MAX];   //单个字符的可能取值
*Tri;
void build_tri()  //创建根节点

     Tri=(Tritree *)malloc(sizeof(Tritree));
     for(int i=0;i<MAX;i++)
          Tri->next[i]=NULL;

void insert_tri(char a[])   //把每个字符串插入字典树中

     Tritree *q,*p=Tri;
     for(int i=0;i<strlen(a);i++)
     
          int id=a[i]-'a';    //取a的第i个字符
          if(p->next[id]==NULL)
          
               q=(Tritree *)malloc(sizeof(Tritree));
               q->count=1;     //第一次赋初值1；
               for(int j=0;j<MAX;j++)
               
                    q->next[j]=NULL;
               
               p->next[id]=q;
               p=p->next[id];
          
          else
          
               p->next[id]->count++;  //计算到达此节点的字的个数，即共同前缀的个数
               p=p->next[id];
          
     

int find_tri(char a[]) //查找统计单词个数

     Tritree *p=Tri;
     for(int i=0;i<strlen(a);i++)
     
          int j=a[i]-'a';
          if(p->next[j]==NULL)  //若没有以此字符串a为前缀的单词，则返回0；
             return 0 ;
          else
          p=p->next[j];
     
     return p->count;  //因为a最终遍历到的地方就是以a为前缀的单词个数；

int main()

   //freopen("input.txt","r",stdin);
   build_tri();
   int flag=0;
   while(gets(a)&&a[0]!='\\0')   //单词输入，gets()为输入一行
     insert_tri(a);
   while(scanf("%s",a)!=EOF)  //前缀输入
   
             printf("%d\\n",find_tri(a));
   
   return 0;