【Trie树】HDU 1671—— Phone List

来源：点击打开链接

好久不发解题报告了……这个是裸的Trie树模板题。需要增加内存释放删除的函数，否则会报MLE。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <stdlib.h>
using namespace std;

char tar[10005][20];

struct Trie
{
    Trie *next[10];
    int v;   //根据需要变化,10,26,52,62
};

Trie *root;

void init()
{
    root=(Trie *)malloc(sizeof(Trie));
    for(int i=0;i<10;i++)
    {
        root->next[i]=NULL;
    }
}

void Trieinsert(char *str)
{
    int len = strlen(str);
    Trie *p = root, *q;
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'0';
        if(p->next[id] == NULL)
        {
            q = (Trie *)malloc(sizeof(Trie));
            q->v = 1;
            for(int 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 Triefind(char *str)
{
	int len = strlen(str);
	Trie *p = root;
	for(int i=0; i<len; ++i)
	{
		int id = str[i]-'0';  //根据需要选择是减去'0'还是'a'，或者是'A'
		p = p->next[id];
		if(p == NULL)
			return 0;
		if(p->v == -1)
			return -1;
	}
	return -1;
}

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

int main()
{
    int testcase,tarnum;
    cin>>testcase;
    while(testcase--)
    {
        char tmp[20];
        init();
        cin>>tarnum;
        int find=0;
        for(int i=0;i<tarnum;i++)
        {
            scanf("%s",tmp);
            if(Triefind(tmp)==-1)
            {
                find=1;
            }
            if(find==1)
                continue;
            Trieinsert(tmp);
        }

        if(find==1)
            cout<<"NO"<<endl;
        else
            cout<<"YES"<<endl;
        Triedel(root);

    }
    return 0;
}