hdu1251 统计难题 数据结构之Trie树

哈哈，第一次用Trie树，从网上找了个模板改了改，然后自己写了search()。

题目很简单，就不解释了。

#include<stdio.h>#include <string.h>const int sonnum=26, base='a';struct Trie{int num;  //记录有多少个单词能到达次，也即相同前缀的个位//bool terminal; //判断是否是结束节点struct Trie *son[sonnum];Trie(){num=1;// terminal=false;memset(son,NULL,sizeof(son));}};void Insert(Trie *root, char *s){Trie *temp=root;while(*s){if(temp->son[*s-base]==NULL)   //不存在 则建立temp->son[*s-base]=new Trie();//NewTrie();else {temp->son[*s-base]->num++;}temp=temp->son[*s-base];s++;}//temp->terminal=true;  //到达尾部，标记一个串}int search(Trie *root,char a[]){Trie *temp=root;int i=0;while(a[i]){if(temp->son[a[i]-base]!=NULL)temp=temp->son[a[i]-base];else return 0;i++;}return temp->num;}int main(){char a[11];Trie *root=new Trie();while(gets(a)){if(a[0]=='\0')break;Insert(root,a);}while(gets(a))printf("%d\n",search(root,a));return 0;}