hdu 1251 统计难题

https://vjudge.net/problem/HDU-1251

题意：略

思路：

经典的字典树例题，模板用上啦，见算法学习汇总。

还有就是经过这题学习到了如何判断以空行结束，那就是gets(s),s[0] == '' 为真就是以空行结束啦。以及hdu提交不要用g++，会mle。

代码：

#include <string.h>
#include <stdio.h>

using namespace std;

const int maxn = 26;

struct trie
{
    trie *next[maxn];

    int flag;

    trie()
    {
        flag = 1;
        memset(next,NULL,sizeof(next));
    }
}*root;

void Insert(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 = new trie();
            p->next[id] = q;
            p = p -> next[id];
        }
        else
        {
            p = p->next[id];

            ++(p->flag);
        }
    }
}

int query(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->flag;
}

void Free(trie* T)
{
    if (T == NULL) return;

    for (int i = 0;i < maxn;i++)
    {
        if (T->next[i]) Free(T->next[i]);
    }

    delete(T);
}

int main()
{
    char s[15];

    root = new trie();

    while (gets(s))
    {
        if (s[0] == '') break;
        Insert(s);
    }

    while (scanf("%s",s) != EOF)
    {
        int ans = query(s);

        printf("%d
",ans);
    }

    Free(root);

    return 0;
}