HDU 1251 统计难题

字典树又一基本题

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 1027

struct node
{
    int count;
    node *next[];
}*root;

char ss[];

node *create()
{
    node *p;
    p = (node *)malloc(sizeof(node));
    p->count = ;
    for(int i=;i<;i++)
        p->next[i] = NULL;
    return p;
}

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

void insert(char *ss)
{
    int i=,k;
    node *p = root;
    while(ss[i])
    {
        k = ss[i++] - 'a';
        if(p->next[k] == NULL)
            p->next[k] = create();
        else
            p->next[k]->count++;
        p = p->next[k];
    }
}

int search(char *ss)
{
    int i=,k,j;
    node *p = root;
    int now = ;
    while(ss[i])
    {
        k = ss[i++] - 'a';
        p = p->next[k];
        if(p == NULL)
            return ;
        else
            now = p->count;
    }
    return now;
}

int main()
{
    int i=,j,k;
    root = create();
    while(gets(ss)&&ss[]!='\0')
    {
        insert(ss);
    }
    i=;
    while(scanf("%s",ss)!=EOF)
    {
        printf("%d\n",search(ss));
    }
    release(root);
    return ;
}