hdu1251 统计难题（字典树）

题目点我点我点我

模版式的字典树，没啥好说的，只是hdu有个巨坑，用G++提交会boom内存，改用C++提交就AC……一开始改了N遍，快抓狂的时候点了一下讨论区，发现都这样or2……还是不懂为神马

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <algorithm>
using namespace std;
#define maxn 26
typedef struct Trie
{
    int v;
    Trie *next[maxn];
}Trie;
Trie *root;

void creattrie(char *str)
{
    Trie *p=root,*q;
    int len=strlen(str);
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)
        {
            q=(Trie *)malloc(sizeof(Trie));
            for(int j=0;j<26;j++)
                q->next[j]=NULL;
            q->v=1;
            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 deltrie(Trie *t)
{
    if(t==NULL)
        return 0;
    for(int i=0;i<maxn;i++)
    {
        if(t->next[i]!=NULL)
            deltrie(t->next[i]);

    }
    free(t);
    return 0;
}


int main()
{
    int i;
    char str[11];
    root=(Trie *)malloc(sizeof(Trie));
    for(i=0;i<maxn;i++)
        root->next[i]=NULL;
    while(gets(str)&&str[0]!='\0')
        creattrie(str);
    memset(str,0,sizeof(str));
    while(~scanf("%s",str))
    {
        int ans=findtrie(str);
        printf("%d\n",ans);
    }
    deltrie(root);

    return 0;
}