PowerOJ 1100 最短的名字 字典树半个模版

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
• Raw

1
3
aaaaa
bbb
abababab

• Sample Output
• Raw

5

字典树记录半个模版

#include <bits/stdc++.h>
using namespace std;
const int N = 5e5+10;
int name[N][26];
int sz;
int n;
int ans;
char s[100000];
int val[N*10];  ///开到足够大
void inssert()
{
    int u = 0;
    int len = strlen(s);
    for(int i = 0;i < len;i++){
        int c = s[i]-'a';
        if(!name[u][c]){
            memset(name[sz],0,sizeof name[sz]);
            name[u][c] = sz++;
        }
        val[u]++;
        u = name[u][c];
    }
    val[u]++;
}

void dfs(int u,int dep)
{
    for(int i = 0;i < 26;i++){
        if(name[u][i]){
            int t = name[u][i];
            if(val[t] == 1){
                ans += dep;
                continue;
            }
            dfs(name[u][i],dep+1);
        }
    }
}

int main(void)
{
    int T;
    scanf("%d",&T);
    while(T--){
        sz = 1;  ///置1
        ans = 0;
        memset(name[0],0,sizeof name[0]);
        memset(val,0,sizeof val);
        scanf("%d",&n);
        for(int i = 1;i <= n;i++){
            scanf("%s",s);
            inssert();
        }
        dfs(0,1);
        printf("%d\n",ans);
    }
    return 0;
}