首页 > 代码库 > CSU OJ 1115 最短的名字(字典树——湖南省第八届大学生计算机程序设计竞赛)

CSU OJ 1115 最短的名字(字典树——湖南省第八届大学生计算机程序设计竞赛)




1115: 最短的名字

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 141  Solved: 56
[Submit][Status][Web Board]

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;
typedef struct node //节点的结构体
{
    int count;//计数
    struct node * next[26];//存储的数组(按照题意来,可以变化的)
}node;
//char s[1010][10010];//这里是看别人这样写,把一个大数组分成二维数组来写(感觉内存浪费比较多)
char s[1000001];
node * newnode() //创建新节点
{
    node *q;
    q=(node*)malloc(sizeof(node));
     q->count=1;
    for(int i=0;i<26;i++)
        q->next[i]=NULL;
    return q;
}
void build(node *T,char *s) //建立字典树
{
    node *p;
    p=T;
    int len,k;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        k=s[i]-'a';
        if(p->next[k]==NULL)
        {
            p->next[k]=newnode();
           // p->count=1; //先前把这个语句写在这里,答案就一直不对,后来写到创建新节点的函数里面就对了
            p=p->next[k];
            /*node *NewNode = (node*) malloc (sizeof (node)); //这里是另一种写法,在用的时候再临时创节点
            NewNode->count = 1;
            for (int j = 0;j<26;j++) NewNode->next[j] = NULL;
            p->next[k] = NewNode;
            p = NewNode;*/
        }
        else
            {
                p=p->next[k];
                p->count++;
            }
    }
}
int search(node *T) //查询字典树
{
    node *q;
    int sum=0;
    q=T;
    for(int i=0;i<26;i++)
    {
        if(T->next[i]!=NULL)
        {
            q=T->next[i];
            sum+=q->count;
            if(q->count>1)
            {
                sum+=search(q);
            }
        }
    }
    return sum;
}
void Release(node *T) //释放内存
{
    for(int i=0;i<26;i++)
        if(T->next[i]!=NULL)
            Release(T->next[i]);
        //delete T; 
        free(T);
}
int main()
{
    int t,n;
    scanf("%d",&t);
    while(t--)
    {
        node *T;
        T=(node *)malloc(sizeof(node));
        T->count=0;
        for(int i=0;i<26;i++)
            T->next[i]=NULL;
        scanf("%d\n",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%s",s);
            build(T,s);
            //gets(s[i]); //二维数组和数组的两种不同的写法
            //build(T,s[i]);
        }
        printf("%d\n",search(T));
        Release(T);
    }
}