【Trie模板】HDU1251-统计难题

【题意】

n统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

【思路】

裸题，不过G++好像会超内存，C++就不会。

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 struct Trie
 {
     int sum;
     Trie *next[];
     Trie()
     {
         sum=;
         for (int i=;i<;i++) next[i]=NULL;
     }
 };
 Trie *root=new Trie;
 char str[];

 void insert()
 {
     int len=strlen(str);
     Trie *tmp=root;
     for (int i=;i<len;i++)
     {
         if (tmp->next[str[i]-'a']==NULL)
         {
             Trie *newnode=new Trie;
             tmp->next[str[i]-'a']=newnode;
         }
         tmp=tmp->next[str[i]-'a'];
         tmp->sum++;
     }
 }

 int find()
 {
     int len=strlen(str);
     Trie *tmp=root;
     for (int i=;i<len;i++)
     {
         if (tmp->next[str[i]-'a']==NULL) return ;
         tmp=tmp->next[str[i]-'a'];
     }
     return tmp->sum;
 }

 void release(Trie *root)
 {
     for (int i=;i<;i++)
         if (root->next[i]!=NULL) release(root->next[i]);
     free(root);
 }

 int main()
 {
     while (gets(str),strcmp(str,"")!=) insert();
     while (scanf("%s",str)!=EOF) cout<<find()<<endl;
     release(root);
     return ;
 }