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

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

1
3
aaaaa
bbb
abababab

5

#include <bits/stdc++.h>
using namespace std;
const int N = 5e5+10;
int name
[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;
}