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<stdio.h> #include<algorithm> #include<iostream> #include<cstring> using namespace std; const int maxn=500005; int T,n,tot,k,f[maxn][27]; char s[10005]; int dfs(int x,int deep) { if (f[x][0]==1) return deep; int ans=0; for (int i=1;i<=26;i++) if (f[x][i]) ans+=dfs(f[x][i],deep+1); return ans; } int main() { scanf("%d",&T); while (T--) { tot=0; memset(f[0],0,sizeof(f[0])); scanf("%d",&n); for (int i=1;i<=n;i++) { scanf("%s",s); for (int j=k=0;s[j];j++) { if (!f[k][s[j]-'a'+1]) { f[k][s[j]-'a'+1]=++tot; memset(f[tot],0,sizeof(f[tot])); } k=f[k][s[j]-'a'+1]; f[k][0]++; } } printf("%d\n",dfs(0,0)); } }