CSU 1115: 最短的名字（字典树）

1115: 最短的名字

 

题目链接：http://acm.csu.edu.cn/OnlineJudge/problem.php?id=1115

Description

在一个奇怪的村子中，很多人的名字都很长，比如aaaaa, bbb and abababab。

名字这么长，叫全名显然起来很不方便。所以村民之间一般只叫名字的前缀。比如叫'aaaaa'的时候可以只叫'aaa'，因为没有第二个人名字的前三个字母是'aaa'。不过你不能叫'a'，因为有两个人的名字都以'a'开头。村里的人都很聪明，他们总是用最短的称呼叫人。输入保证村里不会有一个人的名字是另外一个人名字的前缀（作为推论，任意两个人的名字都不会相同）。

如果村里的某个人要叫所有人的名字（包括他自己），他一共会说多少个字母？

Input

输入第一行为数据组数T (T<=10)。每组数据第一行为一个整数n(1<=n<=1000)，即村里的人数。以下n行每行为一个人的名字（仅有小写字母组成）。输入保证一个村里所有人名字的长度之和不超过1,000,000。

 

Output

对于每组数据，输出所有人名字的字母总数。

Sample Input

1

3

aaaaa

bbb

abababab

Sample Output

5

HINT

 

Source

湖南省第八届大学生计算机程序设计竞赛

 

字典树查找，只要找到前面重叠层某个字符的个数为1或找到啦此字符串的最后一个字符，就返回当前的字符的长度（即树的层数）。。。。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;

struct node
{
    node *a[27]; //存放当前字符后面那一个字符的存放的地址
    int p;  //记录此字符在这一层的个数
}*t,*h;

char s[1005][1005];
int i,j,t0,n,l;

void make(int k,node* h) //建树
{
    h->p++; 计算s[k][j]字符的个数；
    if(k>l) return ;
    if(h->a[s[i][k]-'a']!=NULL) make(k+1,h->a[s[i][k]-'a']);
    else  
    {
        int x,j;
        for(x=k;x<=l;x++)
        {
            t=new node;   //创建新结点
            t->p=1;
            for(j=0;j<26;j++) t->a[j]=NULL;
            h->a[s[i][x]-'a']=t; //保存s[i][x]字符后面那一个字符的存放的地址
            h=t;       
        }
    }
    return;
}

void find(int k,int m,node* h)  //查找
{
    t0++;
    if(k==strlen(s[m])-1||h->p==1) return ;
    find(k+1,m,h->a[s[m][k+1]-'a']);
}

void release(struct node* now) //释放内存
{
    if(now==NULL) return ;
    for(int i=0;i<26;i++)
    {
        if(now->a[i])
        {
            release(now->a[i]);
            now->a[i]=NULL;
        }
    }
    free(now);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        h=new node;  //先创建一个结点
        h->p=0;
        for(i=0;i<26;i++) h->a[i]=NULL; 
        for(i=1;i<=n;i++)
        {
            scanf("%s",s[i]);
            l=strlen(s[i])-1;
            make(0,h);
        }
        int sum=0;
        for(i=1;i<=n;i++)
        {
            t0=0;
            find(0,i,h->a[s[i][0]-'a']);
            sum+=t0;
        }
        release(h);
        printf("%d\n",sum);
    }
    return 0;
}