hdu 1251 统计难题 (字典树)

来源：互联网发布：纸箱计算公式软件编辑：程序博客网时间：2024/04/26 03:11

题目链接: hdu 1251

题目大意: 给出单词的词典，然后有限次询问

每次询问给出的字符在词典中作为前缀的次数

解题思路: 建立词典的字典树

用w标记此结点在建树过程中访问的次数，每经过一次就+1

查询时把查询的字符遍历字典树，遍历最后结点的w值既是答案

代码:

#include <stdio.h>#include <string.h>#include <string.h>#define MAX 1000000struct snode{    char ch1;    int next;   //第二种写法}Tree[MAX];char ch[20];int pd,Index,num[MAX],pre[MAX];void Add_edge(int Star,char ch2)   //建立有向图{    Tree[Index].ch1=ch2,Tree[Index].next=pre[Star];    num[Index]++;    pre[Star]=Index++;}void DFS(int Star,int len,int Tlen)  //DFS插入结点{    int i;    if(len>Tlen)        return ;    for(i=pre[Star];i!=-1;i=Tree[i].next)    {        if(Tree[i].ch1==ch[len-1])        {            num[i]++;            if(len<Tlen)            {                DFS(i,len+1,Tlen);            }            return ;        }    }    Add_edge(Star,ch[len-1]);    DFS(Index-1,len+1,Tlen);}void DFS2(int Star,int len,int Tlen)   //DFS查询w值{    int i;    if(len>Tlen)        return ;    for(i=pre[Star];i!=-1;i=Tree[i].next)    {        if(Tree[i].ch1==ch[len-1])        {            if(len==Tlen)            {                pd=num[i];  //num[ ]是w值                return ;            }            if(len<Tlen)            {                DFS2(i,len+1,Tlen);            }            return ;        }    }    }int main(){    int n=0,i;    Index=1;    memset(num,0,sizeof(num));    memset(pre,-1,sizeof(pre));    memset(Tree,0,sizeof(Tree));    Tree[0].ch1='\0';    Tree[0].next=-1;    while(gets(ch))    //输入词典    {        if(ch[0]=='\0')            break;        DFS(0,1,strlen(ch));    }    while(gets(ch))    //查询    {        pd=0;        DFS2(0,1,strlen(ch));        printf("%d\n",pd);    }    return 0;}