HDU 1251

用字典树统计个数。

一定要交C++。G++MLE。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
using namespace std;
struct Trie
{
    Trie *next[26];
    int count;
    Trie()
    {
        count=0;
        for(int i=0;i<26;i++) next[i]=NULL;
    }
};
void insert(Trie *s,char x[])
{
    int k;
    for(int i=0;x[i];i++)
    {
        k=x[i]-'a';
        if(s->next[k]==NULL)
            s->next[k]=new Trie;
        s=s->next[k];
        ++s->count;
    }
}
int find(Trie *s,char x[])
{
    int k;
    for(int i=0;x[i];i++)
    {
        k=x[i]-'a';
        if(s->next[k]) s=s->next[k];
        else return 0;
    }
    return s->count;
}
int main()
{
    int num=0;
    Trie* root=new Trie;
    char s[20];
    while(gets(s),strcmp(s,""))
    {
        insert(root,s);
    }
    while(gets(s)!=NULL)
    {
        printf("%d
",find(root,s)) ;
    }
}