【Trie模板】HDU1251-统计难题

【题意】

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

【思路】

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

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

void insert()
{
    int len=strlen(str);
    Trie *tmp=root;
    for (int i=0;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=0;i<len;i++)
    {
        if (tmp->next[str[i]-'a']==NULL) return 0;
        tmp=tmp->next[str[i]-'a'];
    }
    return tmp->sum;
}

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

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