HDU 1671 Phone List 字典树-Trie数

题目地址： http://acm.hdu.edu.cn/showproblem.php?pid=1671

一道变形的字典树题目，

直接判断每个字符串出现的次数是不是为1次就行。

记得每次做完了 要释放root，不然会超内纯的哦~~~

代码如下：

#include <iostream>
#include <vector>
#include <List>
#include <deque>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <cmath>
using namespace std;

const int N=11;
typedef long long LL;

char xh[10010][11];

struct Trie
{
    Trie *next[N];//next是表示每层有多少种类的数，如果只是小写字母，则26即可，若改为大小写字母，则是52，若再加上数字，则是62了，这里根据题意来确定。
    int v;//v可以表示一个字典树到此有多少相同前缀的数目，这里根据需要应当学会自由变化。
}*root;

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

void createTrie(char *str)
{
    int i,j,id;
    int len=strlen(str);
    Trie *p=root,*q;
    for(i=0;i<len;i++)
    {
        id=str[i]-'0';//这个地方依据题意来定
        if(p->next[id]==NULL)
        {
            q=(Trie *)malloc(sizeof(Trie));
            q->v=1;//如果没有的话，初始为1
            for(j=0;j<N;j++)
                q->next[j]=NULL;
            p->next[id]=q;
            p=p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p=p->next[id];
        }
    }
}

int findTrie(char *str)
{
    int i,id;
    int len=strlen(str);
    Trie *p=root;
    for(i=0;i<len;i++)
    {
        id=str[i]-'0';
        p=p->next[id];
        if(p==NULL)     //若为空集，表示不存以此为前缀的串
            return 0;
    }
    return p->v;      //此串是字符集中某串的前缀
}

void dealTrie(Trie *T)
{//若果有多组数据-多棵树，那么要释放内纯
    int i;
    if(T==NULL)
        return ;
    for(i=0;i<N;i++)
        if(T->next[i]!=NULL)
            dealTrie(T->next[i]);
    free(T);//释放
}

int main()
{
    int i,j,T;
    cin>>T;
    while(T--)
    {
        int m;
        cin>>m;
        setroot();
        for(i=0;i<m;i++)
        {
            scanf("%s",xh[i]);
            createTrie(xh[i]);
        }
        int flag=0;
        for(i=0;i<m;i++)
        {
            int cnt=findTrie(xh[i]);
            if(cnt!=1)
            {
                flag=1;
                break;
            }
        }
        if(flag!=1)
            printf("YES\n");
        else
            printf("NO\n");
        dealTrie(root);//这个地方如果有多个trie树的话，最好要释放。
    }
    return 0;
}