最短的名字+csuoj+前缀树（字典树）

1115: 最短的名字

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 319   Solved: 115
[ 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

解决方案：此题开始用后缀数组来做，可惜爆内存了，用了轻巧的字典树来写，记录字典树中字母节点上单词的个数，然后再次逐个查询，若出现该节点上只有一个单词，这该长度的前缀就是所求的前缀，累加所有符合的最短的前缀长度就是结果。

code：
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
char a[1002][10001],p[55];
struct Trie
{
    Trie *child[26];
    Trie *fail; ///失败指针
    //int num;
    int cnt; ///由于同一单词可能出现多次。。。
    int v;
    Trie()
    {
        ///num=0;
        cnt=0;
        memset(child,0,sizeof(child));
    }///初始化字典树
};
Trie *root,*s,*lrelia;
int sum;
void Create(char *str)
{
    s=root;
    int i=0;
    while(str[i])
    {
        int id=str[i]-'a';
        if(s->child[id]==0)
        {
            s->child[id]=new Trie;
        }
        s->cnt++;
        s=s->child[id];
        i++;
        //cout<<"sd";
    }
    ///以防出现多个单词
}
void query(char *str)
{

    s=root;
    int i=0;
    while(str[i])
    {
        int id=str[i]-'a';
        if(s->cnt==1)
        {  // cout<<i<<endl;
            sum+=(i);
            return ;
        }
       // cout<<"sd";
        s=s->child[id];
        i++;
    }

}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        root=new Trie();
        scanf("%d",&n);
        for(int i=0; i<n; i++)
        {
            scanf("%s",a[i]);
            //printf("%s\n",a[i]);
            Create(a[i]);
        }
       // cout<<"sd";
        sum=0;
        for(int i=0; i<n; i++)
        {  //cout<<a[i]<<endl;
            query(a[i]);
        }
        printf("%d\n",sum);
    }
    return 0;
}