PKU2001(Shortest Prefixes)字符串匹配-Trie树

/*********************************************************
题目大意：
给出很多字符串,要求找出唯一能识别这个单词的前缀,否则输出该字符串本身;

算法思想：
建立一个Trie树,设置一个变量num,计算每个字符出现的次数;
查找的时候依次从根输出字符串的字符;
如果该结点的下一个结点的num为1;
则说明下一个字符是第一次出现过的字符;
即当前已经输出的为唯一的前缀,直接返回即可;
**********************************************************/
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<climits>
#include<algorithm>
using namespace std;

const int MAX=26;
const int N=1010;
const int M=25;
struct Trie //Trie结点声明
{
    bool isStr;//标记该结点处是否构成一个串
    int num;//统计字符出现的次数
    Trie *next[MAX];//一个指针数组，存放着指向各个儿子节点的指针
};

void insert(Trie *root,const char *s) //将单词s插入到字典树中
{
    if(root==NULL||*s=='\0')
        return;
    Trie *p=root;
    while(*s)
    {
        if(p->next[*s-'a']==NULL)//如果不存在存储该字符的节点，则建立结点
        {
            Trie *temp=new Trie;
            for(int i=0; i<MAX; i++)
            {
                temp->next[i]=NULL;
            }
            temp->isStr=false;
            temp->num=1;
            p->next[*s-'a']=temp;
            p=p->next[*s-'a'];
        }
        else
        {
            p=p->next[*s-'a'];
            p->num++;
        }
        s++;//让指针s指向下一个字符
    }
    p->isStr=true;//单词结束的地方标记此处可以构成一个串
}

void search(Trie *root,const char *s)//查找某个单词s是否已经存在
{
    Trie *p=root;
    while(p&&*s)
    {
        if(p->next[*s-'a']->num==1)
        {
            printf("%c",*s);
            return;
        }
        else
            printf("%c",*s);
        p=p->next[*s-'a'];
        s++;
    }
}

void del(Trie *root)//释放整个字典树占的堆区空间
{
    for(int i=0; i<MAX; i++)
    {
        if(root->next[i]!=NULL)
        {
            del(root->next[i]);
        }
    }
    delete root;
}

int main()
{
    //freopen("C:\\Users\\Administrator\\Desktop\\kd.txt","r",stdin);
    char s[N][M];
    Trie *root = new Trie;
    for(int i=0; i<MAX; i++)
    {
        root->next[i]=NULL;
    }
    root->isStr=false;
    root->num=1;
    int n=0;
    while(gets(s[n]))
    {
        insert(root,s[n]);     //先建立字典树
        n++;
    }
    int t=0;
    while(t<n)
    {
        printf("%s ",s[t]);
        search(root,s[t]);
        printf("\n");
        t++;
    }
    del(root); //释放空间很重要
    return 0;
}