hdu 1251 统计难题 字典树

传送门：hdu1251统计难题
中文题目不做过多的解释

解题思路

已经接近于字典树的模板题目，每次把字典中的单词加入到树里面。
先是要创建这棵树，
1. 线定义结构体，先看我们定义的结构体

typedef struct Trie{    int v;    Trie *next[26];}Trie;

1. 创建字典树函数，字典树的创建如下图：
   ，
   这个图对应的过程就是，
   
   • 首先是一个根节点，他是所有单词的根源root
   • 每个节点都有一个公共前缀，越往后公共前缀包含的字母数越多，叶子节点就是单词本身
   • 比如说第一个叶子节点和第二个叶子节点，他们有公共的前缀是in
   • 因为每一层的字母是肯定不会出现重复的，所以查询的时候是按照一层进行查找的
     创建一个字典树的过程大致如下：

void createTrie(char *str){    int len = strlen(str);    Trie *p = &root, *q;    for(int i=0; i<len; ++i)    {        int id = str[i]-'a';//把这个字符串变为0~25的数字        if(p->next[id] == NULL)//如果这个字母在某一层不存在，就要创建它        {            q = (Trie *)malloc(sizeof(root));            q->v = 1;//初始化前缀为1            for(int j=0; j<26; ++j)//置空它的所有子节点                q->next[j] = NULL;            p->next[id] = q;            p = p->next[id];//创建完毕，让父层节点指向这个新加入的        }        else        {            p->next[id]->v++;//如果存在的话，前缀+1            p = p->next[id];//继续向下走        }    }}

1. 在树种查找一个单词，就是按照输入字母的每一位进行再树中的每一层中进行查找，大致过程如下：

int findTrie(char *str){    int len = strlen(str);    Trie *p = &root;    for(int i=0; i<len; ++i)    {        int id = str[i]-'a';        p = p->next[id];//如果存在一直找到叶子节点        if(p == NULL)//不存在就返回            return 0;    }    return p->v;}

AC代码

#include<cstdio>#include<cstring>#include<cstdlib>#include<algorithm>#include<cmath>using namespace std;typedef struct Trie{    int v;    Trie *next[26];}Trie;Trie root;void createTrie(char *str){    int len = strlen(str);    Trie *p = &root, *q;    for(int i=0; i<len; ++i)    {        int id = str[i]-'a';        if(p->next[id] == NULL)        {            q = (Trie *)malloc(sizeof(root));            q->v = 1;            for(int j=0; j<26; ++j)                q->next[j] = NULL;            p->next[id] = q;            p = p->next[id];        }        else        {            p->next[id]->v++;            p = p->next[id];        }    }}int findTrie(char *str){    int len = strlen(str);    Trie *p = &root;    for(int i=0; i<len; ++i)    {        int id = str[i]-'a';        p = p->next[id];        if(p == NULL)            return 0;    }    return p->v;}int main(){    char str[15];    int i;    for(i=0; i<26; ++i)        root.next[i] = NULL;    while(gets(str) && str[0]!='\0')        createTrie(str);    memset(str, 0, sizeof(str));    while(scanf("%s", str) != EOF)    {        int ans = findTrie(str);        printf("%d\n", ans);    }    return 0;}