Trie树

#include<iostream>
#include<string.h>
#include<stdlib.h>
#include<stdio.h>
using namespace std;
#define MAX 26//这是代表26个字母，如果包含数字需要重新定义

struct Trie
{
    Trie *next[MAX];
    int v;//v可以表示一个字典树到此有多少相同前缀的数目，这里根据需要应当学会自由变化。
};
Trie *root;

void creatTrie(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<MAX;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=strlen(str);
    Trie *p=root;
    for (int 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;////此串是字符集中某串的前缀
}
//比如911为窜已存储，输入9133判断911是不是他的前窜
//当911到达第二个1的时候，p->next[id](2)=NULL

int dealTrie(Trie *T)//删除树
{
    int i;
    if(T==NULL) return 0;
    for (i=0;i<MAX;i++)
    {
        if(T->next[i]!=NULL) dealTrie(T->next[i]);
    }
    free(T);
    return 0;
}

int main()
{
    int T;
    char str[100];
    scanf("%d",&T);
    while(T--)
    {
        int n,i;
        int flag=0;
        root=(Trie *)malloc(sizeof(Trie));//刚开始需要建立结点
        for (i=0;i<MAX;i++)
             root->next[i]=NULL;
        scanf("%d",&n);
        scanf("%s",str);
        creatTrie(str);
        for (i=1;i<n;i++)
        {
             scanf("%s",str);
             if(flag) continue;
             if(findTrie(str)==-1) flag=1;
             else creatTrie(str);
        }
        if(flag) printf("NOn");
        else printf("YESn");
        dealTrie(root);
    }
}