POJ 2001 Shortest Prefix

字典树基本题。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdlib>

using namespace std;
#define N 1027

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

char ss[N][];

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];
    }
}

void search(char *ss)
{
    int i = ,k,j,flag = ;
    node *p = root;
    while(ss[i] && !flag)
    {
        k = ss[i++] - 'a';
        p = p->next[k];
        if(p->count == )
        {
            flag = ;
            printf("%s ",ss);
            for(j=;j<i;j++)
                printf("%c",ss[j]);
            printf("\n");
        }
    }
    if(!flag)
        printf("%s %s\n",ss,ss);
}

int main()
{
    int i=,j,k;
    root = create();
    while(scanf("%s",ss[i])!=EOF)
    {
        insert(ss[i]);
        i++;
    }
    for(j=;j<i;j++)
        search(ss[j]);
    release(root);
    return ;
}