HDU 1671

这是一道字典树的问题，直接利用模板，不过还要一个释放内存的函数！

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
typedef struct trie
{
    int v;
    trie *next[10];
};


trie *root;


void creattrie(char *str)
{
    int len,i,j;
    trie *p=root,*q;
    len=strlen(str);
    for(i=0; i<len; i++)
    {
        int id=str[i]-'0';
        if(p->next[id]==NULL)
        {
            q=(trie *)malloc(sizeof(trie));
            q->v=1;
            for(j=0; j<10; j++)
                q->next[j]=NULL;
            p->next[id]=q;
            p=p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p=p->next[id];
        }
    }
    p->v=-1;
}


int findtrie(char *str)
{
    int len,i;
    trie *p=root;
    len=strlen(str);
    for(i=0; i<len; i++)
    {
        int id=str[i]-'0';
        p=p->next[id];
        if(p==NULL)
            return 0;
        if(p->v==-1)
            return -1;//前面有一个串是此串的前缀
    }
    return -1;//此串是前面一个串的前缀！


}


int deal(trie *T)
{
    int i;
    if(T==NULL)return 0;
    for(i=0; i<10; i++)
    {
        if(T->next[i]!=NULL)
        {
            deal(T->next[i]);
        }
    }
    free(T);
    return 0;
}


int main()
{
    int t,num;
    int i;
    int flag;
    char str[20];
    scanf("%d",&t);
    while(t--)
    {
        flag=0;
        root=(trie *)malloc(sizeof(trie));
        for(i=0; i<10; i++)
            root->next[i]=NULL;
        scanf("%d",&num);
        for(i=0; i<num; i++)
        {
            scanf("%s",str);
            if(findtrie(str))
                flag=1;
            if(flag)
                continue;
            creattrie(str);
        }
        if(flag)
        {
            printf("NO\n");
        }
        else
        {
            printf("YES\n");
        }
        deal(root);
    }
    return 0;
}