HDU 1671(Trie树）

题目链接

题意：t组测试，每组n个数，每个数都不能是其他数的前缀

思路：简单的Trie。但是因为是多组测试，一定一定一定一定要把每次的树free掉

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<malloc.h>

using namespace std;

typedef struct Trie_node
{
    int flag;
    struct Trie_node* next[10];
}TrieNode,*Trie;

char s[10000][11];
Trie T;
int n;

bool create(Trie &T)
{
    T=(Trie)malloc(sizeof(TrieNode));
    for(int i=0;i<10;i++)
        T->next[i]=NULL;
    for(int j=0;j<n;j++)
    {
        int len=strlen(s[j]);
        Trie p=T;
        int i=0;
        while(i<len&&p->next[s[j][i]-'0']!=NULL)
        {
            p=p->next[s[j][i]-'0'];
            if(p->flag==1)
                return false;
            p->flag=0;
            i++;
        }
        if(i==len)
            return false;
        Trie q;
        while(i<len)
        {
            q=(Trie)malloc(sizeof(TrieNode));
            q->flag=0;
            for(int j=0;j<10;j++)
                q->next[j]=NULL;
            p->next[s[j][i]-'0']=q;
            p=q;
            i++;
        }
        p->flag=1;
    }
    return true;
}

void delTrie(Trie &T){
    if(T == NULL)
        return;
    for(int i = 0; i<=9; i++)
    {
        if(T->next[i]!=NULL)
            delTrie(T->next[i]);
    }
    free(T);
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        getchar();
        int flag1=0;
        for(int i=0;i<n;i++)
            gets(s[i]);
        if(create(T))
            printf("YES\n");
        else
            printf("NO\n");
        delTrie(T);   //一定要释放内存！！
    }
    return 0;
}