湖南省第八届大学生程序设计大赛原题 E

E - 最短的名字

Time Limit:5000MS     Memory Limit:65536KB     64bit IO Format:%lld & %llu

Submit Status Practice CSU 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 

分析：字典树的模板题

#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
#define MAX 28
struct node{
    int flag;//标记是否为节点
    int num;;//表示一个字典树到此有多少相同前缀的数目
    struct node *next[MAX];
};

void Insert(node *root,char *str){
    if(root==NULL&&*str=='\0')
        return ;
    node *p=root;
    int last;//记录上一次经过的节点的num
    while(*str!='\0'){
        if(p->next[*str-'a']==NULL){
            node *temp=new node;
            for(int i=0;i<MAX;i++)
                temp->next[i]=NULL;
            temp->flag=0;
            temp->num=0;
            p->next[*str-'a']=temp;
        }
        p=p->next[*str-'a'];
        p->num+=1;
        str++;
    }
    p->flag=1;
}

void Delete(node *root){
    for(int i=0;i<MAX;i++)
        if(root->next[i]!=NULL)
            Delete(root->next[i]);
    delete (root);
}

int Query(node *root,int cnt){
    int sum=0;
    if(root->num==1)
        return cnt;
    for(int i=0;i<MAX;i++)
        if(root->next[i]!=NULL)
            sum+=Query(root->next[i],cnt+1);
    return sum;
}

int main(){
    char str[1000005];
    int T,n;
    cin>>T;
    while(T--){
        node *root=new node;
        for(int i=0;i<MAX;i++)
            root->next[i]=NULL;
        root->flag=0;
        cin>>n;
        for(int i=0;i<n;i++){
            cin>>str;
            Insert(root,str);
        }
        cout<<Query(root,0)<<endl;
        Delete(root);
    }
    return 0;
}